基于YoloV4的绝缘子目标检测程序源代码+数据集+训练模型,人工智能课程设计作业。文件结构
.
│ predict.py # 对图片进行预测
│ train.py # 训练模型
│ voc_annotation.py # 对VOC数据集处理导出索引
│ yolo.py # 预测程序的子程序
│
├─img # 存放预测后的图像
├─logs # 存放训练的模型文件
├─model_data # 存放预训练模型
│ new_classes.txt # 类别的名称
│ yolo_anchors.txt # 先验框的大小
│
├─nets # 网络结构
│ CSPdarknet.py # CSPdarkNet53主干特征网络
│ yolo4.py # FPN、SPP等网络
│ yolo_training.py # 模型训练子程序
│
├─utils # 数据加载、NMS等
│ dataloader.py # 数