【文件属性】:
文件名称:Pytorch-Faster-RCNN
文件大小:16KB
文件格式:ZIP
更新时间:2021-02-26 06:32:53
Python
Faster-RCNN的实现
准备
先决条件:Python> = 3.6和Pytorch> = 1.6
安装
数据集应为VOC格式,数据目录应如下所示
|---your dataset name
|---Annotations
|---JPEGImages
训练模型
sh train.sh
推论模型
python demo.py
【文件预览】:
Pytorch-Faster-RCNN-main
----coco_utils.py(9KB)
----engine.py(4KB)
----train.py(4KB)
----utils.py(10KB)
----train.sh(175B)
----dataset()
--------transform_data.py(216B)
--------Dataset.py(3KB)
----demo.py(2KB)
----README.md(532B)
----coco_eval.py(12KB)
----transforms.py(2KB)