使用mmdetection框架训练自己的数据集
根据以下教程即可:
https://blog.csdn.net/qq_35077107/article/details/124768392
修改数据集的类别后发现报错
AssertionError: The `num_classes` (1) in CenterNetHead of MMDataParallel doe
检查了发现类别数目其实都已经修改好了,为什么还会报错呢,其实是因为你没有重新编译环境,具体方法是;python setup.py install。 详见:https://blog.csdn.net/sunnyblogs/article/details/121409301
mmdetection 配置文件解读
https://www.cnblogs.com/Allen-rg/p/12196888.html
关于图片尺寸等问题
https://zhuanlan.zhihu.com/p/141176981
值得注意的是:是把你自己数据集里面的图片放缩并自适应(默认keep_ratio = TRUE)到(1333,800)
可视化结果的自定义(锚框,文字等)
mmdet/core/visualization/image.py在这里面,重新定义了:
def imshow_det_bboxes:
bbox_color = None
text_color = None
font_size = 0
thickness = 0