Update script.py
Browse files
script.py
CHANGED
|
@@ -64,7 +64,7 @@ def run_inference(model, image_path, conf_threshold, save_path):
|
|
| 64 |
|
| 65 |
if __name__ == "__main__":
|
| 66 |
|
| 67 |
-
TEST_IMAGE_PATH =
|
| 68 |
SUBMISSION_SAVE_PATH = "submission.csv"
|
| 69 |
CONF_THRESHOLD = 0.30
|
| 70 |
|
|
|
|
| 64 |
|
| 65 |
if __name__ == "__main__":
|
| 66 |
|
| 67 |
+
TEST_IMAGE_PATH = "/tmp/data/test_images"
|
| 68 |
SUBMISSION_SAVE_PATH = "submission.csv"
|
| 69 |
CONF_THRESHOLD = 0.30
|
| 70 |
|