sharvari0b26
/

phase2a-2

Model card Files Files and versions

sharvari0b26 commited on Jan 9

Commit

431142b

·

verified ·

1 Parent(s): ab8099b

Update submission/script.py

Files changed (1) hide show

submission/script.py +1 -1

submission/script.py CHANGED Viewed

@@ -60,7 +60,7 @@ def run_inference(model, image_path, conf_threshold, save_path):
 if __name__ == "__main__":
-    TEST_IMAGE_PATH = r"rf-detr\dataset\test"
     SUBMISSION_SAVE_PATH = "submission.csv"
     CONF_THRESHOLD = 0.30

 if __name__ == "__main__":
+    TEST_IMAGE_PATH = "/tmp/data/test_images"
     SUBMISSION_SAVE_PATH = "submission.csv"
     CONF_THRESHOLD = 0.30