1、简介
这里主要是针对Light Head RCNN进行了行人检测的训练,Backbone是基于EfficientNet,其结构如下:
2、项目结构
项目结构如下:
训练集如下:
测试结果如下:
对应文件说明:
- 1、Train.py用来训练
- 2、Webcam.py用来测试视频或者摄像头
- 3、weights文件夹下是EfficientNet不同结构的预训练权重
注意:训练时候数据集路径根据实际情况进行修改,pycharm可以直接一键运行训练和测试。
这里主要是针对Light Head RCNN进行了行人检测的训练,Backbone是基于EfficientNet,其结构如下:
项目结构如下:
训练集如下:
测试结果如下:
对应文件说明:
注意:训练时候数据集路径根据实际情况进行修改,pycharm可以直接一键运行训练和测试。