File "vit_h_maskdecoder.pth" is missing

ymy-k / Hi-SAM

[arXiv preprint] Hi-SAM: Marrying Segment Anything Model for Hierarchical Text Segmentation

Apache License 2.0

193 stars 10 forks source link

File "vit_h_maskdecoder.pth" is missing #1

Closed tatue64 closed 7 months ago

tatue64 commented 7 months ago

I tried to run the Hi-SAM Visualization Demo of hierarchical segmentation using the following command

python demo_hisam.py --checkpoint pretrained_checkpoint/hi_sam_l.pth --model-type vit_l --input demo/img293.jpg --output demo/ --hier_det

but got the error FileNotFoundError: [Errno 2] No such file or directory: 'pretrained_checkpoint/vit_h_maskdecoder.pth'

I did not find any reference to this file in the README. Do you provide this file for download anywhere?

ymy-k commented 7 months ago

I tried this command again just now, but I got no error. Are you sure that you use Hi-SAM but not SAM-TSS? SAM-TSS is only a stroke level segmentation model.

tatue64 commented 7 months ago

Thanks for the quick reply!

You are right,: I used the tss checkpoint instead of the one of hi_sam.

It now works!