faster-rcnn环境搭建

1.下载地址:https://github.com/bubbliiiing/faster-rcnn-pytorch
2.环境搭建过程参考yolov8地址:https://www.cnblogs.com/czeyu/p/18068440
3.将下载后的代码包解压后打开,配置解释器
4.进行测试,下载权重文件训练所需的voc_weights_resnet.pth或者voc_weights_vgg.pth以及主干的网络权重可以在百度云下载。
voc_weights_resnet.pth是resnet为主干特征提取网络用到的;
voc_weights_vgg.pth是vgg为主干特征提取网络用到的;
链接: https://pan.baidu.com/s/1S6wG8sEXBeoSec95NZxmlQ
提取码: 8mgp
VOC数据集下载地址如下,里面已经包括了训练集、测试集、验证集(与测试集一样),无需再次划分:
链接: https://pan.baidu.com/s/1-1Ej6dayrx3g0iAA88uY5A
提取码: ph32
5.下载完库后解压,在百度网盘下载frcnn_weights.pth,放入model_data,运行predict.py,输入img/street.jpg后点击enter键即可

6.出现如下结果即为成功

ps:遇到的问题AttributeError: ‘ImageDraw‘ object has no attribute ‘textsize‘
查看pillow版本,如果大于10,则执行pip install Pillow==9.4.0降低pillow版本即可

posted @ 2024-03-30 21:00  czeyu  阅读(266)  评论(0)    收藏  举报