# 简介

该测试用例用于PyTorch目标检测模型Fasterrcnn测试。

# 运行
*  train.py中get_dataset函数需要根据实际数据集情况设置json文件的位置、类别数目等。
## 单卡
    python3 train.py  --batch-size=2 -j 8 --epochs=26 --data-path=/path/to/datasets/folder --output-dir=/path/to/result/save/folder
## 单机多卡
    mpirun -np 4 --hostfile hostfile --bind-to none `pwd`/single_process.sh localhost
## 多机多卡
    mpirun -np $np --hostfile hostfile --bind-to none `pwd`/single_process.sh ${master_ip}

# 参考
[https://github.com/pytorch/vision/tree/master/references/detection](https://github.com/pytorch/vision/tree/master/references/detection)