上传者: 42150360
|
上传时间: 2022-05-06 22:45:27
|
文件大小: 5.31MB
|
文件类型: ZIP
Retinanet:目标检测模型在Pytorch当中的实现
目录
性能情况
训练数据集
权值文件名称
测试数据集
输入图片大小
mAP 0.5:0.95
mAP 0.5
VOC07+12
VOC-Test07
600x600
-
81.56
所需环境
torch==1.2.0
文件下载
训练所需的retinanet_resnet50.pth可以在百度云下载。
链接:
提取码: krr5
VOC数据集下载地址如下:
VOC2007+2012训练集
链接: 提取码: eiw9
VOC2007测试集
链接: 提取码: dsda
预测步骤
a、使用预训练权重
下载完库后解压,在百度网盘下载retinanet_resnet50.pth,放入model_data,运行predict.py,输入
img/street.jpg
利用video.py可进行摄像头检测。
b、使用自己训练的权重
按照训练