Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ excavator_path = Path('.')
|
|
16 |
example_path = excavator_path / "Examples"
|
17 |
|
18 |
# Get example filepaths in a list of lists
|
19 |
-
example_list = [["
|
20 |
|
21 |
def predict(img, conf_thres=0.35, iou_thres=0.5): # default : conf(0.25), iou(0.7)
|
22 |
model = YOLO('./weights/excavator_only_best_v8m.pt')
|
|
|
16 |
example_path = excavator_path / "Examples"
|
17 |
|
18 |
# Get example filepaths in a list of lists
|
19 |
+
example_list = [["Examples/" + example] for example in os.listdir(example_path)]
|
20 |
|
21 |
def predict(img, conf_thres=0.35, iou_thres=0.5): # default : conf(0.25), iou(0.7)
|
22 |
model = YOLO('./weights/excavator_only_best_v8m.pt')
|