下载数据集

在这个网址下载数据集: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
Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐