#286 · 主分类: 计算机视觉

icevision

ai annotation-parsers coco-dataset coco-parser computer-vision deep-learning effecientdet fastai faster-rcnn mask-rcnn object-detection pycocotools python pytorch pytorch-lightning tutorials voc-dataset voc-parser

一个与训练库无关的计算机视觉框架 - 可插拔到任何训练库:Fastai、Pytorch-Lightning以及更多即将推出

项目最后更新:11/26/24

GitHub Stars

867

Forks数量

146

贡献者数量

41

许可证

Apache-2.0

收录理由

IceVision 不把你锁死在某个训练库上。它把 COCO、VOC 等标注格式统一到一套连贯的 Python 接口里,自带解析器和数据变换,训练流程可交给 Fastai 或 PyTorch-Lightning,因此更换框架时不必重写数据加载代码。来自多个模型库的预训练权重也通过同一套接口调用,比较不同架构变得相当省事。内置的数据集检查面板和自动清理工具能在训练开始前发现错误标注和文件问题,面对杂乱的真实数据能省下不少时间。它还支持多任务训练,检测、分割、分类这些任务头可以一起跑,适合研究团队以及想搭建自定义检测流程、又不愿手工拼装各种组件的人。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目