22
33## 介绍
44
5- 本目录包含了采用MegEngine实现的经典[ RetinaNet] ( https://arxiv.org/pdf/1708.02002> ) 网络结构,
6- 同时提供了在COCO2017数据集上的完整训练和测试代码。
5+ 本目录包含了采用MegEngine实现的经典[ RetinaNet] ( https://arxiv.org/pdf/1708.02002> ) 网络结构,同时提供了在COCO2017数据集上的完整训练和测试代码。
76
87网络的性能在COCO2017验证集上的测试结果如下:
98
10- | 模型 | mAP<br >@5-95 | batch<br >/gpu | gpu | speed<br >(8gpu) | speed<br >(1gpu)|
11- | --- | --- | --- | --- | --- | --- |
12- | retinanet-res50-1x-800size | 36.0 | 2 | 2080 | 2.27(it/s) | 3.7(it/s) |
9+ | 模型 | mAP<br >@5-95 | batch<br >/gpu | gpu | speed<br >(8gpu) | speed<br >(1gpu) |
10+ | --- | --- | --- | --- | --- | --- |
11+ | retinanet-res50-coco- 1x-800size | 36.0 | 2 | 2080ti | 2.27(it/s) | 3.7(it/s) |
1312
14- * MegEngine v0.3 .0
13+ * MegEngine v0.4 .0
1514
1615## 如何使用
1716
1817模型训练好之后,可以通过如下命令测试单张图片:
1918
2019``` bash
21- python3 tools/inference.py -f retinanet_res50_1x_800size .py \
20+ python3 tools/inference.py -f retinanet_res50_coco_1x_800size .py \
2221 -i ../../assets/cat.jpg \
2322 -m /path/to/retinanet_weights.pkl
2423```
@@ -35,8 +34,8 @@ python3 tools/inference.py -f retinanet_res50_1x_800size.py \
3534
3635## 如何训练
3736
38- 1 . 在开始训练前,请确保已经下载解压好[ COCO数据集 ] ( http://cocodataset.org/#download ) ,
39- 并放在合适的数据目录下,准备好的数据集的目录结构如下所示(目前默认使用coco2017的数据集 ):
37+ 1 . 在开始训练前,请确保已经下载解压好[ COCO2017数据集 ] ( http://cocodataset.org/#download ) ,
38+ 并放在合适的数据目录下,准备好的数据集的目录结构如下所示(目前默认使用COCO2017数据集 ):
4039
4140```
4241/path/to/
@@ -46,14 +45,14 @@ python3 tools/inference.py -f retinanet_res50_1x_800size.py \
4645 | |val2017
4746```
4847
49- 2 . 准备预训练的` backbone ` 网络权重:可使用 megengine.hub 下载` megengine ` 官方提供的在ImageNet上训练的resnet50模型 , 并存放在 ` /path/to/pretrain.pkl ` 。
48+ 2 . 准备预训练的` backbone ` 网络权重:可使用 megengine.hub 下载` megengine ` 官方提供的在ImageNet上训练的ResNet-50模型 , 并存放在 ` /path/to/pretrain.pkl ` 。
5049
51503 . 在开始运行本目录下的代码之前,请确保按照[ README] ( ../../../README.md ) 进行了正确的环境配置。
5251
53524 . 开始训练:
5453
5554``` bash
56- python3 tools/train.py -f retinanet_res50_1x_800size .py \
55+ python3 tools/train.py -f retinanet_res50_coco_1x_800size .py \
5756 -n 8 \
5857 --batch_size 2 \
5958 -w /path/to/pretrain.pkl
@@ -65,7 +64,7 @@ python3 tools/train.py -f retinanet_res50_1x_800size.py \
6564- ` -n ` , 用于训练的devices(gpu)数量,默认使用所有可用的gpu.
6665- ` -w ` , 预训练的backbone网络权重的路径。
6766- ` --batch_size ` ,训练时采用的` batch size ` , 默认2,表示每张卡训2张图。
68- - ` --dataset-dir ` , coco数据集的根目录 ,默认` /data/datasets/coco ` 。
67+ - ` --dataset-dir ` , COCO2017数据集的上级目录 ,默认` /data/datasets ` 。
6968
7069默认情况下模型会存在 ` log-of-retinanet_res50_1x_800size ` 目录下。
7170
@@ -74,18 +73,18 @@ python3 tools/train.py -f retinanet_res50_1x_800size.py \
7473在训练的过程中,可以通过如下命令测试模型在` COCO2017 ` 验证集的性能:
7574
7675``` bash
77- python3 tools/test.py -n 8 \
78- -f retinanet_res50_1x_800size.py \
76+ python3 tools/test.py -f retinanet_res50_coco_1x_800size.py \
77+ -n 8 \
7978 --model /path/to/retinanet_weights.pt \
80- --dataset_dir /data/datasets/coco
79+ --dataset_dir /data/datasets
8180```
8281
8382` tools/test.py ` 的命令行选项如下:
8483
8584- ` -f ` , 所需要测试的网络结构描述文件。
8685- ` -n ` , 用于测试的devices(gpu)数量,默认1;
8786- ` --model ` , 需要测试的模型;可以从顶部的表格中下载训练好的检测器权重, 也可以用自行训练好的权重。
88- - ` --dataset_dir ` ,coco数据集的根目录 ,默认` /data/datasets `
87+ - ` --dataset_dir ` ,COCO2017数据集的上级目录 ,默认` /data/datasets `
8988
9089## 参考文献
9190
0 commit comments