mmdetection训练widerface数据集(不需要转换)
·
下载数据集
在这个网址下载数据集:https://drive.google.com/drive/folders/1kyh7yu1LmfPwigseA4XvgdMqTCRhmATY
在这个网址下载数据集的annotation:https://github.com/sovrasov/wider-face-pascal-voc-annotations
按照下图这样的格式整理好,不然会报错:
安装好环境
网址有详细的安装教程:GET STARTED — MMDetection 3.1.0 documentation
修改配置文件
1.找到configs/wider_face/ssd300_8xb32-24e_widerface.py,打开就能看到是需要打开4个文件夹查看对应的配置文件。
路径在configs/_base_
_base_ = [
'../_base_/models/ssd300.py', '../_base_/datasets/wider_face.py',
'../_base_/default_runtime.py', '../_base_/schedules/schedule_2x.py'
]
修改数据集存放路径data_root/修改batch_size小一点,不然显存不够,num_workers(window系统要改成0,不然跑不了)/
data_root = '/home/lixinru/Vir_env/mmdetection-main/data/WIDERFace/'
batch_size=16, num_workers=2, dataset=dict(pipeline=train_pipeline))

2.修改epoch,也就是训练的轮数,一般先训练200.
max_epochs=200

3.其他2个文件默然参数就行
开始训练
python tools/train.py configs/wider_face/ssd300_8xb32-24e_widerface.py
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)