CenterNet-Lite
CenterNet的PyTorch版本(以对象为点)。我只支持resnet18版本。没有DLA或沙漏版本。
我已经在VOC0712和COCO 2017上进行了培训。您可以从BaiDuYunDisk下载它们:
链接: :
密码:jz4q
官方CenterNet充分利用了DCN,而我只是将其替换为YOLOv3中使用的SPP,因为我有点懒惰〜
在VOC上:
数据
地图
(官方)resnet18 + DCN
VOC2007
75.7
(我们的)resnet18 + SPP
VOC2007
75.3
在COCO上:
数据
美联社
AP50
(官方)resnet18 + DCN
COCO测试版
28岁
44.9
(我们的)resnet18 + SPP
可可值
25.8
45.4
我仍在尝试一些新的方法来增强CenterNet-Lite的强度。
安装
2022-03-09 20:01:59
165KB
Python
1