自带的分类和模型的的对应关系
coco_classes.txt yolo4.weight.pth
voc_classes.txt yolo4_voc_weigths.pth
TS_classes.txt Epoch100-Total_Loss3.8129-Val_Loss5.0751.pth
自带predict.py调用nets/yolo4.py
train.py做训练模型使用
参数说明:
lr:学习率
Batch_size:批处理尺寸,几张图片更新一次参数
Init_Epoch:起始记书数
Freeze_Epoch:训练到多少次冻结一下
TS_detection.py 做模型的输出测试,调用yolo.py文件(其中包含模型参数)
yolo.py中关键参数:
anchors_path :目标匡的尺寸
classes_path:类别
model_image_size:模型尺寸固定,不能变动
confidence:置信度
iou:交并比
letterbox_image:暂时无用
通过标注生成的图片参数信息:
filename :图片名称
path:图片位置
width:宽度
height:高度
de