文件名称:CenterNet-better:CenterNet的易于理解和更高性能的版本
文件大小:212KB
文件格式:ZIP
更新时间:2024-04-09 11:37:29
computer-vision deep-learning object-detection Python
中心网 此仓库基于我的dl_lib实现,我的dl_lib中的某些代码部分基于 。 动机 是我在对象检测领域中最喜欢的论文之一。但是,其有点难以理解。我相信CenterNet可以得到更高的分数,并且可以以更优雅的方式实现,因此我编写了此仓库。 表现 此回购使用较少的训练时间来获得更好的性能,与旧的回购相比,它花费了近一半的训练时间并获得了1〜2 pts的更高的mAP。这是性能表。 骨干 地图 第一人称射击 V100 FPS 训练模型 ResNet-18 29.8 92 113 ResNet-50 34.9 57 71 ResNet-101 36.8 43 50 什么是新的? 将配置视为对象。您可以运行配置文件并检查config值,这对于调试确实很有帮助。 默认情况下,常见的培训/测试脚本。您只需要在操场上调用dl_train/test --num-gpus x ,您的项目只