
Gen-EfficientNet-Pytorch: 预先训练的EfficientNet及EfficientNet-Lite等模型
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
Gen-EfficientNet-Pytorch是一个包含预先训练好的EfficientNet和EfficientNet-Lite模型的库,适用于PyTorch框架。这些高效模型在多个图像分类任务中表现出色。
PyTorch的EfficientNets(包括EfficientNet、MixNet、MobileNetV3等)提供了一种“通用”实现方式,涵盖了从MobileNet V1/V2块序列派生出的大多数计算/参数高效架构,其中包括通过自动神经架构搜索发现的一些架构。所有这些模型都是由GenEfficientNet或MobileNetV3类来实现,并且使用基于字符串的体系结构定义来进行配置以确定块布局。
在2020年8月19日更新中,新增了我用timm训练得到的一个更优版本的PyTorch EfficientNet-B3砝码(准确率为82.1% top-1)。此外还添加了一个经过PyTorch训练有素的EfficientNet-Lite0模型(top-1准确率75.5%)。
更新内容还包括ONNX和Caffe2导出/实用程序脚本,使其能够与最新的PyTorch及ONNX版本兼容。另外新增了基于ONNX运行时验证脚本激活功能,在大多数情况下可以实现与timm等效项的相同效果。
全部评论 (0)
还没有任何评论哟~


