augmentedstartups / AS-One

Easy & Modular Computer Vision Detectors, Trackers & SAM - Run YOLOv9,v8,v7,v6,v5,R,X in under 10 lines of code.
https://www.augmentedstartups.com
GNU General Public License v3.0
608 stars 104 forks source link

How to set a custom path for downloaded weights ? #12

Closed franklinkemta closed 1 year ago

franklinkemta commented 2 years ago

Hello guys, Thank you for such amazing work!

Is there a way to specify a custom path for downloaded weights instead of the current working dir ? e.g: to be able to use the Tracker this way

Tracker(asone.BYTETRACK, model, use_cuda=False, custom_model_path)

in case the custom_model_path already exists the .pts are loaded from that path (the same way it's downloaded to the working dir at the moment).

Thanks

MNMaqsood commented 1 year ago

5