yellowdolphin commited on
Commit
5816e16
·
1 Parent(s): c6447ed

debug cropped

Browse files
Files changed (1) hide show
  1. app.py +7 -1
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
- cropped = sorted(glob(f'{working}/results_ensemble/crops/*/{Path(yolo_source).name}'))
 
 
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