bubbliiiing目标检测代码仓库:https://github.com/bubbliiiing?tab=repositories
Train.py
- classes_path 数据集类别
- model_path 预训练权重
- save_dir 保存路径
- train_annotation_path 训练路径
- Val_annotation_path 验证路径
get_map.py
- classes_path 数据集类别
- VOCdevkit 数据集文件夹
- map_out_path 结果输出文件夹
YOLO/detr/ssd.py
- model_path 训练好的权重
- classes_path 类别
utils_map.py
- fig.canvas.set_window_title -->fig.convas.manager.set_window_title