kevinconka commited on
Commit
12a0f0a
·
verified ·
1 Parent(s): 4624c16

Update model_yolov5.py

Browse files
Files changed (1) hide show
  1. model_yolov5.py +1 -0
model_yolov5.py CHANGED
@@ -16,6 +16,7 @@ def load_model(model_path, img_size=640):
16
  def inference(model, image):
17
  """Run inference on image and return annotated image."""
18
  results = model(image, size=model.img_size)
 
19
  annotator = Annotator(np.asarray(image))
20
  for *box, _, cls in reversed(results.pred[0]):
21
  # label = f'{model.names[int(cls)]} {conf:.2f}'
 
16
  def inference(model, image):
17
  """Run inference on image and return annotated image."""
18
  results = model(image, size=model.img_size)
19
+ print(results)
20
  annotator = Annotator(np.asarray(image))
21
  for *box, _, cls in reversed(results.pred[0]):
22
  # label = f'{model.names[int(cls)]} {conf:.2f}'