search

jiawen-zhu / HQTrack

Tracking Anything in High Quality
MIT License
744 stars 65 forks source link

sam_hq_vit_l.pth model not found error #6

Closed MUHAMMEDZEYN closed 1 year ago

MUHAMMEDZEYN commented 1 year ago

Hello,

I was trying to run the demo.py file but I had the error above, so I figured that in the code it is written as vit_l model which is not shared so I changed it as below :

model_type = 'vit_h' sam_checkpoint = os.path.join(base_path, '..', 'segment_anything_hq/pretrained_model/sam_hq_vit_h.pth')

Great job, by the way.

jiawen-zhu commented 1 year ago

Hi~ For convenience, we created the demo with a SAM-HQ_vit_l pretrained model. You can easily replace the Mask Refiner model by modifying the settings in demo.py. The pretrained model can be downloaded from SAM-HQ.