Spaces:
Running
Running
original sources url added
Browse files- backend/engine.py +2 -0
- backend/model.py +1 -0
backend/engine.py
CHANGED
|
@@ -76,6 +76,8 @@ def run(model, video_path, line, config, on_frame, save_annotated=False):
|
|
| 76 |
|
| 77 |
start = time.time()
|
| 78 |
|
|
|
|
|
|
|
| 79 |
results = model.track(
|
| 80 |
source=video_path,
|
| 81 |
tracker="bytetrack.yaml",
|
|
|
|
| 76 |
|
| 77 |
start = time.time()
|
| 78 |
|
| 79 |
+
# https://docs.ultralytics.com/modes/predict/#inference-sources
|
| 80 |
+
# https://docs.ultralytics.com/modes/track/#why-choose-ultralytics-yolo-for-object-tracking
|
| 81 |
results = model.track(
|
| 82 |
source=video_path,
|
| 83 |
tracker="bytetrack.yaml",
|
backend/model.py
CHANGED
|
@@ -21,6 +21,7 @@ def ensure_openvino():
|
|
| 21 |
f'https://huggingface.co/Perception365/VehicleNet-Y26s/resolve/main/weights/best.pt'
|
| 22 |
)
|
| 23 |
|
|
|
|
| 24 |
if not OV_DIR.exists():
|
| 25 |
YOLO(str(PT_PATH)).export(format="openvino", dynamic=True)
|
| 26 |
|
|
|
|
| 21 |
f'https://huggingface.co/Perception365/VehicleNet-Y26s/resolve/main/weights/best.pt'
|
| 22 |
)
|
| 23 |
|
| 24 |
+
# https://docs.ultralytics.com/modes/export/#arguments
|
| 25 |
if not OV_DIR.exists():
|
| 26 |
YOLO(str(PT_PATH)).export(format="openvino", dynamic=True)
|
| 27 |
|