# Fast R-CNN pytorch
This is an implementation of Fast R-CNN using pytorch on the animal images of COCO dataset.
Fast R-CNN uses ROIPooling to avoid repeated calculation in R-CNN and combines classification and location togerther using FC in neural networks.
To prepare data, download and unzip in the COCO2017 folder.
To install required packages, run
```
pip install -r requirements.txt
```
To create regions using selective search, run
```
python script/create_regions.py
```
This may take several hours to complete.
Herein, vgg19_bn is used in Fast R-CNN to extract features.
To train the model, run
```
python train.py --epochs num_epochs --cuda True/False
```
To test the model on specific images, run
```
python test.py --jpg_path jpg_path
```
test sample
![](sample.png)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
利用coco2017数据集训练Fast-RCNN模型(训练过程详细步骤记录): (1)检测数据集利用选择搜索算法(selective-search)生成一定数量的候选框, (2)将候选框与真实标注框进行IOU(交并比)计算,将真是标注框的作为正样本,将0.1<IOU<0.5的当做负样本 (3)设计网络骨干模型,利用VGG19,利用ROIPlooing方法将建议框映射到输出特征层 (4)设置输出为一个分类分支(类别类数+背景类(1))与标注回归分支 (5)设置交叉熵损失与回归损失 (6)训练网络模型
资源推荐
资源详情
资源评论
收起资源包目录
Fast-R-CNN-pytorch-master.rar (41个子文件)
Fast-R-CNN-pytorch-master
dataset
__init__.py 63B
COCOdataset.py 3KB
__pycache__
COCOdataset.cpython-38.pyc 2KB
__init__.cpython-38.pyc 218B
.idea
Fast-R-CNN-pytorch-master.iml 329B
workspace.xml 12KB
misc.xml 304B
inspectionProfiles
profiles_settings.xml 174B
modules.xml 302B
model
fast_rcnn.pkl 533.4MB
requirements.txt 201B
sample.png 350KB
fast_rcnn
__init__.py 88B
fast_rcnn.py 2KB
roipooling.py 1KB
__pycache__
fast_rcnn.cpython-38.pyc 2KB
roipooling.cpython-38.pyc 2KB
__init__.cpython-38.pyc 253B
script
create_region.py 5KB
train.py 5KB
test.py 5KB
README.md 813B
fonts
spanish.ttf 329KB
korean.ttf 1.17MB
chinese_cht.ttf 7.03MB
kannada.ttf 778KB
latin.ttf 54KB
french.ttf 755KB
uyghur.ttf 38KB
german.ttf 755KB
japan.ttc 3.32MB
hindi.ttf 217KB
urdu.ttf 38KB
arabic.ttf 100KB
telugu.ttf 967KB
marathi.ttf 67KB
cyrillic.ttf 55KB
tamil.ttf 139KB
persian.ttf 31KB
simfang.ttf 10.09MB
nepali.ttf 217KB
共 41 条
- 1
郭庆汝
- 粉丝: 2w+
- 资源: 31
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功
- 1
- 2
前往页