sabrinabenas commited on
Commit
64e43c8
β€’
1 Parent(s): 0d79970
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import torchvision
10
  from dlclive import DLCLive, Processor
11
 
12
  from PIL import Image, ImageColor, ImageFont, ImageDraw
13
-
14
  import numpy as np
15
  import math
16
 
@@ -112,6 +112,7 @@ def crop_animal_detections(yolo_results,
112
  likelihood_th):
113
 
114
  ## Extract animal crops
 
115
  list_labels_as_str = yolo_results.names # ['animal', 'person', 'vehicle']
116
  list_np_animal_crops = []
117
  # for every image
 
10
  from dlclive import DLCLive, Processor
11
 
12
  from PIL import Image, ImageColor, ImageFont, ImageDraw
13
+ # check git lfs pull!!
14
  import numpy as np
15
  import math
16
 
 
112
  likelihood_th):
113
 
114
  ## Extract animal crops
115
+ print(yolo_results)
116
  list_labels_as_str = yolo_results.names # ['animal', 'person', 'vehicle']
117
  list_np_animal_crops = []
118
  # for every image