How can i add and eval_ritm argument ?

uncbiag / SimpleClick

SimpleClick: Interactive Image Segmentation with Simple Vision Transformers (ICCV 2023)

MIT License

209 stars 32 forks source link

How can i add and eval_ritm argument ? #2

Closed edmundman closed 1 year ago

edmundman commented 1 year ago

Trying to run the demo and i get this error, the pth is in the main directory.

D:\Code\SimpleClick>python demo.py --checkpoint cocolvis_vit_huge.pth Traceback (most recent call last): File "D:\Code\SimpleClick\demo.py", line 59, in main() File "D:\Code\SimpleClick\demo.py", line 18, in main model = utils.load_is_model(checkpoint_path, args.device, cpu_dist_maps=True) TypeError: load_is_model() missing 1 required positional argument: 'eval_ritm'

qinliuliuqin commented 1 year ago

The issue has been fixed. You only need to add "--eval-ritm" if you are evaluating RITM models. Thank you!