raab/efishSignalDetector
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

bf

Browse Source
This commit is contained in:
Till Raab 2023-10-26 07:30:50 +02:00
parent a98f488813
commit f507877624
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
inference.py
View File

@ -32,7 +32,8 @@ def show_sample(img_tensor, outputs, detection_threshold):
) )
plt.show() plt.show()
def infere_model(test_loader, model, detection_th=0.8) def infere_model(test_loader, model, detection_th=0.8):
pass
if __name__ == '__main__': if __name__ == '__main__':
model = create_model(num_classes=NUM_CLASSES) model = create_model(num_classes=NUM_CLASSES)