Spaces:

sdwalker62
/

Unreal-Engine-5-Tanks-YOLOv5

Runtime error

sdwalker62 commited on Jan 14, 2023

Commit

7eb1a91

•

1 Parent(s): ab0ae16

Update detect.py

Files changed (1) hide show

detect.py CHANGED Viewed

@@ -51,7 +51,7 @@ from utils.torch_utils import select_device, smart_inference_mode
 @smart_inference_mode()
-def run(
         weights=ROOT / 'yolov5s.pt',  # model path or triton URL
         source=ROOT / 'data/images',  # file/dir/URL/glob/screen/0(webcam)
         data=ROOT / 'data/coco128.yaml',  # dataset.yaml path

 @smart_inference_mode()
+def detect(
         weights=ROOT / 'yolov5s.pt',  # model path or triton URL
         source=ROOT / 'data/images',  # file/dir/URL/glob/screen/0(webcam)
         data=ROOT / 'data/coco128.yaml',  # dataset.yaml path