• ww1993922

    运行python3 demo/demo_embed_mask.py报错

    Traceback (most recent call last):
    File "demo/demo_embed_mask.py", line 124, in
    main()
    File "demo/demo_embed_mask.py", line 90, in main
    min_image_size=args.min_image_size
    File "./demo/predictor.py", line 119, in init
    _ = checkpointer.load(cfg.MODEL.WEIGHT)
    File "./fcos_core/utils/checkpoint.py", line 62, in load
    self._load_model(checkpoint)
    File "./fcos_core/utils/checkpoint.py", line 98, in _load_model
    load_state_dict(self.model, checkpoint.pop("model"))
    File "./fcos_core/utils/model_serialization.py", line 80, in load_state_dict
    model.load_state_dict(model_state_dict)
    File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 845, in load_state_dict
    self.class.name, "\n\t".join(error_msgs)))
    RuntimeError: Error(s) in loading state_dict for GeneralizedRCNN:
    size mismatch for rpn.head.proposal_margin_pred.weight: copying a param with shape torch.Size([1, 256, 3, 3]) from checkpoint, the shape in current model is torch.Size([1, 4, 1, 1]).

    发布在 原创分享专区 阅读更多
  • ww1993922

    Traceback (most recent call last):
    File "demo/demo_embed_mask.py", line 124, in
    main()
    File "demo/demo_embed_mask.py", line 90, in main
    min_image_size=args.min_image_size
    File "./demo/predictor.py", line 119, in init
    _ = checkpointer.load(cfg.MODEL.WEIGHT)
    File "./fcos_core/utils/checkpoint.py", line 62, in load
    self._load_model(checkpoint)
    File "./fcos_core/utils/checkpoint.py", line 98, in _load_model
    load_state_dict(self.model, checkpoint.pop("model"))
    File "./fcos_core/utils/model_serialization.py", line 80, in load_state_dict
    model.load_state_dict(model_state_dict)
    File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 845, in load_state_dict
    self.class.name, "\n\t".join(error_msgs)))
    RuntimeError: Error(s) in loading state_dict for GeneralizedRCNN:
    size mismatch for rpn.head.proposal_margin_pred.weight: copying a param with shape torch.Size([1, 256, 3, 3]) from checkpoint, the shape in current model is torch.Size([1, 4, 1, 1]).

    发布在 社区求助区(SOS!!) 阅读更多
  • ww1993922

    File "demo/demo_embed_mask.py", line 8, in
    from demo.predictor import COCODemo
    File "./demo/predictor.py", line 13, in
    from alfred.vis.image.common import get_unique_color_by_id
    ModuleNotFoundError: No module named 'alfred'

    发布在 社区求助区(SOS!!) 阅读更多

与 ManaAI社区 的连接断开,我们正在尝试重连,请耐心等待