SamsungLabs / fbrs_interactive_segmentation

[CVPR2020] f-BRS: Rethinking Backpropagating Refinement for Interactive Segmentation https://arxiv.org/abs/2001.10331

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Assertion Error while running the demo

tshaark opened this issue · comments

Traceback (most recent call last):
File "demo.py", line 59, in
main()
File "demo.py", line 15, in main
checkpoint_path = utils.find_checkpoint(cfg.INTERACTIVE_MODELS_PATH, args.checkpoint)
File "/home/tshaark/fbrs_interactive_segmentation/isegm/inference/utils.py", line 173, in find_checkpoint
assert len(model_checkpoints) == 1
AssertionError

I get this error when I try to run the demo and a similar one when I try to train. Where am I going wrong?

It's most likely that you incorrectly specified the path to the model weights (check --checkpoint argument).

Do I have to download the weights for the models I want to use? If so, what should be path of the weights directory?

@tshaark you can download pre-trained weights and put them anywhere. just give the correct path to --checkpoint argument.
--checkpoint=/path to weights/'