快速使用mobiledets-yolov4

1.下载项目

git clone https://github.com/inacmor/mobiledets-yolov4-pytorch.git

2.准备数据

2.1 准备VOC格式的数据集

put your trainset and labels(xml) in data/Imgs and data/Annotations.

这里选择直接将一个VOC格式的数据集通过软链接链接过来

ln -s /data_jupiter/dataset/209_VOC_new/Annotations ./
ln -s /data_jupiter/dataset/209_VOC_new/JPEGImages ./Imgs 

2.2 修改classes.txt

LuggageVehicle
BridgeVehicle
Plane
RefuelVehicle
Person
FoodVehicle
LuggageVehicleHead
FollowMe
TractorVehicle

3.开始训练

3.0 环境安装

1. 安装pytorch(略)

2. 安装apex

   git clone https://github.com/NVIDIA/apex
   cd apex
   python setup.py install

3.1 训练前准备工作

python ready.py
python kmeans.py

3.2 开始训练

没有预训练模型的话需要将模型加载注释掉,修改train.py

#model.load_state_dict(torch.load(pretrained_weights), strict=False)

# 执行训练
python train.py