Spaces:
Sleeping
Sleeping
Commit
·
5816e16
1
Parent(s):
c6447ed
debug cropped
Browse files
app.py
CHANGED
@@ -69,6 +69,10 @@ def fast_yolo_crop(image):
|
|
69 |
rmtree(working / 'labels', ignore_errors=True)
|
70 |
rmtree(working / 'results_ensemble', ignore_errors=True)
|
71 |
|
|
|
|
|
|
|
|
|
72 |
mpimg.imsave(yolo_source, image)
|
73 |
|
74 |
detect.run(weights=checkpoint_files[4:],
|
@@ -84,7 +88,9 @@ def fast_yolo_crop(image):
|
|
84 |
exist_ok=True,
|
85 |
name=str(working / 'results_ensemble'))
|
86 |
|
87 |
-
|
|
|
|
|
88 |
assert len(cropped) == 1, f'{len(cropped)} maritime species detected'
|
89 |
cropped = cropped[0]
|
90 |
species = Path(cropped).parent.name
|
|
|
69 |
rmtree(working / 'labels', ignore_errors=True)
|
70 |
rmtree(working / 'results_ensemble', ignore_errors=True)
|
71 |
|
72 |
+
print("image:", type(image))
|
73 |
+
print(image.shape)
|
74 |
+
print("yolo_source:", yolo_source)
|
75 |
+
print(type(yolo_source))
|
76 |
mpimg.imsave(yolo_source, image)
|
77 |
|
78 |
detect.run(weights=checkpoint_files[4:],
|
|
|
88 |
exist_ok=True,
|
89 |
name=str(working / 'results_ensemble'))
|
90 |
|
91 |
+
glob_pattern = f'{working}/results_ensemble/crops/*/{Path(yolo_source).name}'
|
92 |
+
print("glob_pattern:", glob_pattern)
|
93 |
+
cropped = sorted(glob(glob_pattern))
|
94 |
assert len(cropped) == 1, f'{len(cropped)} maritime species detected'
|
95 |
cropped = cropped[0]
|
96 |
species = Path(cropped).parent.name
|