Pytorch简单CenterNet-45
如果您正在寻找其他CenterNet,请尝试!
这个仓库是一个简单的pytorch实现的 ,一些代码取自。 顾名思义,此版本简单易读,所有复杂的部分(数据加载器,沙漏,训练循环等)都以更简单的方式重写了。 顺便说一下,还添加了对nn.parallel.DistributedDataParallel的支持,因此该实现的训练速度比正式代码快得多(在8个GPU上约为75 img / s与36 img / s)。
享受!
要求:
python> = 3.5
pytorch == 0.4.1或1.1.0(DistributedDataParallel培训仅在使用1.1.0时可用)
tensorboardX(可选)
入门
禁用cudnn批处理规范化。 打开torch/nn/functional.py并找到与torch.batch_norm的行,并
1