self-made_ava_dataset_tool icon indicating copy to clipboard operation
self-made_ava_dataset_tool copied to clipboard

TypeError: __init__() got an unexpected keyword argument 'pretrained'

Open Cecilia-zwq opened this issue 8 months ago • 0 comments

I got error while following step1 to get JSON file:

(slowfast2) /mnt/workspace/annotation/self-made_ava_dataset_tool> python detectron2_outvia3.py /mnt/workspace/annotation/self-made_ava_dataset_tool/faster_rcnn_r50_fpn_2x_coco.py /mnt/workspace/annotation/self-made_ava_dataset_tool/faster_rcnn_r50_fpn_2x_coco_bbox_mAP-0.384_20200504_210434-a5d8aa15.pth --input /mnt/workspace/Data/shortVideoTrainCut/daria_wave1/*.jpg --gen_via3 --output mnt/workspace/annotation/self-made_ava_dataset_tool/annotations_proposal --score-thr 0.5 --show
Traceback (most recent call last):
  File "/mnt/workspace/annotation/self-made_ava_dataset_tool/detectron2_outvia3.py", line 151, in <module>
    main()
  File "/mnt/workspace/annotation/self-made_ava_dataset_tool/detectron2_outvia3.py", line 62, in main
    model = init_detector(args.config, args.checkpoint, device=args.device)
  File "/root/anaconda3/envs/slowfast2/lib/python3.9/site-packages/mmdet/apis/inference.py", line 66, in init_detector
    model = MODELS.build(config.model)
  File "/root/anaconda3/envs/slowfast2/lib/python3.9/site-packages/mmengine/registry/registry.py", line 570, in build
    return self.build_func(cfg, *args, **kwargs, registry=self)
  File "/root/anaconda3/envs/slowfast2/lib/python3.9/site-packages/mmengine/registry/build_functions.py", line 232, in build_model_from_cfg
    return build_from_cfg(cfg, registry, default_args)
  File "/root/anaconda3/envs/slowfast2/lib/python3.9/site-packages/mmengine/registry/build_functions.py", line 121, in build_from_cfg
    obj = obj_cls(**args)  # type: ignore
TypeError: __init__() got an unexpected keyword argument 'pretrained'

It suggest that I got an unexpectedauguments 'pretrained'

Here's the package version in my env:

detectron2                0.6
mmcv                      2.0.0rc4                 
mmcv-full                 1.7.2            
mmdet                     3.3.0           
mmengine                  0.10.4   
python                    3.9.19
torch                     1.12.0+cu113

I followed your Installation step to install the env packages. After I pip install mmdet, it suggested that I should use mmcv2.2.0rc4 because of the mmdet version is 3.3.0. But in your step, the version of your mmcv seems to be 1.7.2.

I guess maybe the error is caused by the high version of mmdet. Please tell me your version and what else I can do.

Cecilia-zwq avatar Jun 06 '24 03:06 Cecilia-zwq