bf

2023-11-16 13:30:26 +01:00 · 2023-11-16 13:30:26 +01:00 · 0f0128439e
commit 0f0128439e
parent 324c690841
1 changed files with 0 additions and 1 deletions
--- a/inference.py
+++ b/inference.py
@ -54,7 +54,6 @@ def infere_model(inference_loader, model, dataset_name, detection_th=0.8):

        with torch.inference_mode():
            outputs = model(images)
-            # ToDo: save outputs in label folder !

        for image, img_name, output in zip(images, img_names, outputs):
            # x0, y0, x1, y1