Subh775 commited on
Commit
faa7e0c
·
1 Parent(s): 74118fd

original sources url added

Browse files
Files changed (2) hide show
  1. backend/engine.py +2 -0
  2. 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