augmentedstartups / AS-One

Easy & Modular Computer Vision Detectors, Trackers & SAM - Run YOLOv9,v8,v7,v6,v5,R,X in under 10 lines of code.

Home Page:https://www.augmentedstartups.com

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to set a custom path for downloaded weights ?

franklinkemta opened this issue · comments

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