facebookresearch / detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
https://detectron2.readthedocs.io/en/latest/
Apache License 2.0
30.05k stars 7.42k forks source link

LVIS1.0 model weights not found #2349

Open IanPhilips opened 3 years ago

IanPhilips commented 3 years ago

Instructions To Reproduce the Issue:

  1. Full runnable code or full changes you made:

    cfg = get_cfg()
    modelPath="LVISv1-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml"
    # add project-specific config (e.g., TensorMask) here if you're not running a model in detectron2's core library
    cfg.merge_from_file(model_zoo.get_config_file(modelPath))
    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5  # set threshold for this model
    # Find a model from detectron2's model zoo. You can use the https://dl.fbaipublicfiles... url as well
    cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url(modelPath)
    predictor = DefaultPredictor(cfg)
    outputs = predictor(im)
  2. Full logs you observed:

    
    RuntimeError                              Traceback (most recent call last)
in () 5 cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5 # set threshold for this model 6 # Find a model from detectron2's model zoo. You can use the https://dl.fbaipublicfiles... url as well ----> 7 cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url(modelPath) 8 predictor = DefaultPredictor(cfg) 9 outputs = predictor(im) /usr/local/lib/python3.6/dist-packages/detectron2/model_zoo/model_zoo.py in get_checkpoint_url(config_path) 98 suffix = _ModelZooUrls.CONFIG_PATH_TO_URL_SUFFIX[config_path] 99 return _ModelZooUrls.S3_PREFIX + name + "/" + suffix --> 100 raise RuntimeError("{} not available in Model Zoo!".format(name)) 101 102 RuntimeError: LVISv1-InstanceSegmentation/mask_rcnn_R_50_FPN_1x not available in Model Zoo! ``` ## Environment: Detectron2 Tutorial Colab Do y'all plan on adding the lvis1.0 model to the model zoo? I found the config (https://github.com/facebookresearch/detectron2/tree/master/configs/LVISv1-InstanceSegmentation) but it looks like the model weights won't load.
Thanossrs commented 3 years ago

I have the same problem. any idea?

thank you

nagyrajmund commented 3 years ago

Hi, I would also like to use the pretrained LVIS 1.0 instance segmentation models. Do you plan to add them to the model zoo?

Arturs commented 2 years ago

Hi! Is there any progress on this?

XRDevIEEE commented 2 years ago

Yes

IEEE

On Feb 12, 2022, at 1:02 PM, arturs @.***> wrote:  Hi! Is there any progress on this?

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.

Arturs commented 2 years ago

Can you be more specific about the progress?

I am executing: model_name = 'LVISv1-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_1x.yaml' model_zoo.get_checkpoint_url(model_name)

Getting: RuntimeError: Pretrained model for LVISv1-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_1x.yaml is not available!

PS: It works if: model_name='LVISv0.5-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_1x.yaml'