【文件属性】:
文件名称:SA-AutoAug:用于物体检测的可感知规模的自动增强(CVPR 2021)
文件大小:6.19MB
文件格式:ZIP
更新时间:2021-05-13 22:25:40
Python
SA自动
用于物体检测的可感知规模的自动增强
陈育康,李彦伟,陶涛,陆奇,储瑞航,李磊,贾佳雅
[ ] [ ]
该项目为CVPR 2021论文“”提供了实现。 可感知规模的AutoAug提供了新的搜索空间和搜索指标,以找到用于对象检测的有效数据参数化策略。 它是在和。 搜索和培训代码均已发布。 为了便于更多使用,我们重新实现了基于的训练代码。
安装
有关代码,请按照的说明进行操作。
有关代码,请按照的说明进行操作。
对于代码,请按照的说明进行操作。
搜索
(您可以跳过此步骤,直接训练我们搜索到的政策。)
要使用8个GPU进行搜索,请运行:
cd /path/to/SA-AutoAug/maskrcnn-benchmark
export NGPUS=8
python3 -m torch.distributed.launch --nproc_per_node= $NGPUS to