Bootstrap

A ConvNet for the 2020s 代码复现(目标检测)

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

;