常用骨干网络预训练输入尺寸

1. EfficientNet-V1

  • B0:224×224
  • B1:240×240
  • B2:260×260
  • B3:300×300
  • B4:380×380
  • B5:456×456
  • B6:528×528
  • B7:600×600

官方 TensorFlow 源码仓库(Google 官方)

tpu/models/official/efficientnet at master · tensorflow/tpu · GitHub 仓库 README /model_builder.py 直接硬编码各模型训练分辨率:

python 复制代码
def efficientnet_params(model_name):
  """Get efficientnet params based on model name."""
  params_dict = {
      # (width_coefficient, depth_coefficient, resolution, dropout_rate)
      'efficientnet-b0': (1.0, 1.0, 224, 0.2),
      'efficientnet-b1': (1.0, 1.1, 240, 0.2),
      'efficientnet-b2': (1.1, 1.2, 260, 0.3),
      'efficientnet-b3': (1.2, 1.4, 300, 0.3),
      'efficientnet-b4': (1.4, 1.8, 380, 0.4),
      'efficientnet-b5': (1.6, 2.2, 456, 0.4),
      'efficientnet-b6': (1.8, 2.6, 528, 0.5),
      'efficientnet-b7': (2.0, 3.1, 600, 0.5),
      'efficientnet-b8': (2.2, 3.6, 672, 0.5),
      'efficientnet-l2': (4.3, 5.3, 800, 0.5),
  }
  return params_dict[model_name]

2. EfficientNet-V2

  • V2-B0:224×224
  • V2-B1:240×240
  • V2-B2:260×260
  • V2-B3:300×300
  • V2-S:384×384
  • V2-M:480×480
  • V2-L:480×480

3. ResNet 系列(ResNet18/34/50/101/152)

全部 224×224(原版 torchvision 预训练)

论文:Deep Residual Learning for Image Recognition (He et al., CVPR 2016) arXiv:https://arxiv.org/pdf/1512.03385v1.pdf

4. MobileNet

  • MobileNetV2:224×224
  • MobileNetV3-Large / V3-Small:224×224

5. ConvNeXt

  • ConvNeXt-Tiny / Small / Base:224×224

6. Swin Transformer V1

  • Swin-Tiny / Small / Base:224×224

7. ViT(Vision Transformer)

  • ViT-B/16、ViT-L/16:224×224(原版)
  • ViT-B/16-384:384×384(高分辨率版本)
相关推荐
Yolanda_20221 小时前
19.神经网络-最大池化的使用
人工智能·深度学习·神经网络
W***25921 小时前
2026 企业 AI 办公平台选型指南:可完成全链路任务的 AI 工具评估
人工智能
RockHopper20252 小时前
面向工业现实的原生数字化工程框架概要说明
人工智能·智能体·世界模型·工业数字化
红海云2 小时前
Jev:给智能系统做判断的模型
大数据·数据库·人工智能
wjkjpcba2 小时前
PCBA烧录程序是什么:PCBA包工包料厂家解析烧录与测试
linux·数据库·人工智能·smt贴片加工·pcba贴片加工厂
FL16238631292 小时前
智慧医疗X光图像小儿手腕外伤检测数据集VOC+YOLO格式2538张9类别
人工智能·yolo·机器学习
呆萌很2 小时前
消融实验对比方法
人工智能·深度学习
EchoMind-Henry3 小时前
Muse外设两条接入路,成本该怎么算
人工智能·ai
Zootopia6263 小时前
飞行力学知识梳理1|飞行性能与稳定性
人工智能·python·算法·机器学习·无人机·学习方法·信息与通信