A ConvNet for the 2020s 代码复现(目标检测)
GitHub:https://github.com/facebookresearch/ConvNeXt/tree/main/object_detection
ConvNeXt目标检测模型基于Swin-Transformer-Object-Detection模型,只要修改其中的配置文件即可。
Swin-Transformer-Object-Detection复现:
http://t.csdn.cn/Iw0ol
1.训练模型下载
添加模型路径
命令:python tools/train.py configs/convnext/mask_rcnn_convnext_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco_in1k.py
类别更改
configs/base/models/mask_rcnn_convnext_fpn.py
问题:KeyError: “MaskRCNN: ‘ConvNeXt is not in the backbone registry’”
解决:把tools/train.py 移动到根目录
python train.py configs/convnext/mask_rcnn_convnext_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco_in1k.py