search

SamsungLabs / fbrs_interactive_segmentation

[CVPR2020] f-BRS: Rethinking Backpropagating Refinement for Interactive Segmentation https://arxiv.org/abs/2001.10331
Mozilla Public License 2.0
583 stars 94 forks source link

Assertion Error while running the demo #9

Closed tshaark closed 4 years ago

tshaark commented 4 years ago

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?

ksofiyuk commented 4 years ago

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

tshaark commented 4 years ago

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

pkraison commented 4 years ago

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