Faster-Rcnn:Two-Stage目标检测模型在Tensorflow2当中的实现
目录
性能情况
训练数据集
权值文件名称
测试数据集
输入图片大小
mAP 0.5:0.95
mAP 0.5
VOC07+12
VOC-Test07
-
-
77.5
所需环境
tensorflow-gpu==2.2.0
文件下载
训练所需的voc_weights.h5可以去百度网盘下载
链接: 提取码: 92e8
VOC数据集下载地址如下:
VOC2007+2012训练集
链接: 提取码: eiw9
VOC2007测试集
链接: 提取码: dsda
预测步骤
a、使用预训练权重
下载完库后解压,在百度网盘下载voc_weights.h5,放入model_data,运行predict.py,输入
img/street.jpg
利用video.py可进行摄像头检测。
b、使用自己训练的权重
按照训
1