roboflow / zero-shot-object-tracking

Object tracking implemented with the Roboflow Inference API, DeepSort, and OpenAI CLIP.
https://blog.roboflow.com/zero-shot-object-tracking/
GNU General Public License v3.0
357 stars 62 forks source link

Can't get attribute 'DetectionModel' on <module 'models.yolo' #25

Open MZeinab opened 1 year ago

MZeinab commented 1 year ago

I get this error when running: !python clip_object_tracker.py --source ./data/video/videolife.mp4 --detection-engine yolov5 --weights ./models/best2.pt --info

Can't get attribute 'DetectionModel' on <module 'models.yolo' from '/content/zero-shot-object-tracking/models/yolo.py'> xdgb

could you please help

AmartyaCSB commented 1 year ago

@MZeinab, I have the same error. Were you able to solve this?

MZeinab commented 1 year ago

It about a long time ago, I finally solved it. I think the problem was about the new version of yolov5 ... I mean V6. that the original code was not compatible with the new version of yolo. I think I used !git clone -b fix-custom-weights https://github.com/ikitcheng/zero-shot-object-tracking.git
and it worked.

Hope to be correct and useful.