CenterNet-pytorch-lightning:PyTorch Lightning模块附带的CenterNet（对象为点...

带PyTorchLightning的CenterNet 描述 我尝试更辉煌的的清洁实现。 特征 解耦的骨干网和磁头可简化骨干网集成 将样本创建分为图像增强（使用 ）和实际样本创建 随附Lightning模块，但也可以与优质的PyTorch一起使用 剥离了所有未用于重现论文结果的代码 较小的代码库和更有意义的变量名 需要更少的内存 与原始实施结果相同或稍好 待办事项 原始存储库的某些功能尚未实现，但欢迎拉取请求！ 3D边界框检测 ExtremeNet检测 Pascal VOC数据集 怎么跑 首先，安装依赖项 # clone CenterNet git clone https://github.com/tteepe/CenterNet-pytorch-lightning # install CenterNet cd CenterNet-pytorch-lightning pip inst