Faster RCNN 对血液细胞目标检测

目录

1. 介绍

2. 工具函数介绍 utils

2.1 xml 文件的读取 get_label_from_xml

2.2 绘制边界框 draw_bounding_box

2.3 计算 iou 函数 miou

3. dataset 数据加载

4. train 函数

4.1 头文件

4.2 超参数

4.3 数据加载

4.4 建立 faster rcnn 模型

4.5 可视化

4.6 训练

4.7 训练过程的损失

4.8 log 文件信息

5. 预测文件

6. 完整代码

6.1 dataset

6.2 train

6.3 utils

6.4 predict

---

1. 介绍

这次做一个目标检测的内容，通过 Faster RCNN 对血液细胞进行检测

项目地址：Faster RCNN 对血液细胞目标检测

以下是文件的目录：

• data 是项目的数据
• dataset 是数据加载
• train 训练函数的主体
• utils 相关的工具函数
• predict 是预测文件
• log 是训练日志信息

 

所有数据的文件结构如图所示，jpg 是原始图像，xml是label 的标签

单个样本的展示如下：

左边是数据 jpg 格式的图像 ，右边是 xml 文件格式的描述性信息

 

最后生成的结