文件名称：CenterNet-pytorch-lightning:PyTorch Lightning模块附带的CenterNet（对象为点-Zhou，Xingyi等）的重构实现
文件大小：85KB
文件格式：ZIP
更新时间：2024-04-07 09:47:25
object-detection pose-estimation centernet Python
带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
【文件预览】：
CenterNet-pytorch-lightning-main
----.github()
--------dependabot.yml(429B)
--------workflows()
----tests()
--------utilities.py(741B)
--------data()
--------test_transforms.py(2KB)
--------test_sample_encode_decode.py(2KB)
--------__init__.py(0B)
--------requirements.txt(96B)
--------test_train_detection.py(1KB)
--------test_models.py(1KB)
----LICENSE(11KB)
----setup.cfg(905B)
----CenterNet()
--------models()
--------centernet_detection.py(14KB)
--------centernet_pose.py(7KB)
--------decode()
--------centernet.py(4KB)
--------transforms()
--------centernet_detection_test.py(2KB)
--------utils()
----requirements.txt(199B)
----setup.py(428B)
----.gitignore(1KB)
----README.md(3KB)