deepvac是DeepVAC规范的标准库。这个版本仍然是一个技术预览版,但已经很成熟了。我们准备在十月前发布第一个正式版本:1.0。相比今天发布的技术预览版,1.0版本主要带来的变化将是:TNN转换器、MNN转换器、QAT训练范式、文档完善、bugfix。下面是0.5.8版本的发布说明:
- core模块
- aug模块
- backbones模块
- cast模块
- datasets模块
- loss模块
- experimental模块
core模块
- 添加PyTorch标准训练、DDP训练、AMP训练、EMA训练;
- 添加checkpoint加载、预训练模型加载、jit模型加载;
- 支持tensorboard;
- 添加3种report报告(FaceReport、OcrReport、ClassifierReport);
aug模块
- 添加多种基础类型的aug算子;
- 添加多种用于语义分割领域的aug算子;
- 添加多种用于人脸检测领域的aug算子;
- 添加多种用于目标检测领域的aug算子;
- 添加多种用于OCR领域的aug算子;
- 添加多种Composer。
backbones模块
- 添加resnet、mobilenet、regnet、repvgg backbone;
- 添加多种可复用的网络模块;
- 添加resnet50 benchmark模块;
cast模块
- 支持TorchScript模块的script、trace;
- 支持onnx;
- 支持ncnn;
- 支持CoreML;
- 支持TensorRT;
- 支持动态量化;
- 支持静态量化。
datasets模块
- 添加FileLineDataset、FileLineCvStrDataset、FileLineCvSegDataset;
- 添加OsWalkDataset
- 添加CocoCVDataset
loss模块
- 添加MaskL1Loss、DiceLoss、BalanceCrossEntropyLoss、BCEBlurWithLogitsLoss、FocalLoss、QFocalLoss、WingLoss
- 添加用于人脸检测的MultiBoxLoss
- 添加ArcFace、CurricularFace
experimental模块
- 添加PyTorch蒸馏训练