平民化、轻量级深度学习视觉库 vikit-learn 更新 v0.0.2

第二届开放原子大赛ICT领域专区赛4大AI赛项，120万奖金池，技术专家辅导，立即报名，等你来战！

2025年1月3日，轻量级深度学习视觉库vikit-learn版本更新v.0.0.2。

本次更新内容包括：

• 优化joint模型的损失函数；
• 增加用于调教ocr的数据集工具；
• 增加用于ocr训练的数据转换器；
• 接入timm库，支持mobilenet-v4模型的backbone；
• 优化带有位置信息的下采样层；
• 优化onnx模型导出功能；

Vikit-learn是一个使用Python开发的、基于深度学习技术的计算机视觉处理工具包。

该软件包旨在提供一系列易于使用的工具来处理现实世界的任务。

该项目仍在积极建设和开发中，敬请期待本作！

目前支持：图像分类、目标检测、语义分割、关键点和关节检测、OCR

 

 

 

 

Dependencies

• matplotlib>=3.7.5
• torch>=2.4.0
• torchvision>=0.19.0
• torchmetrics>=1.4.2
• lightning-utilities>=0.11.7
• faster-coco-eval>=1.6.0
• pycocotools>=2.0.7
• opencv-python>=4.10.0
• flet>=0.24.0
• shapely>=2.0.6
• tqdm>=4.66.5
• timm>=1.0.12