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

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(高分辨率版本)
相关推荐
irislinAmelia3 小时前
缺陷检测AI关键点在哪
人工智能·嵌入式硬件·1024程序员节
Ai-_Man3 小时前
Meta Muse还是不可以导出左侧所有对话,只能点进去单条内容导出,方法不对吗?
开发语言·人工智能·ai·小程序·c#·1024程序员节
安托智造3 小时前
MBD:从二维图纸到三维模型的制造革命(下)
大数据·人工智能·制造·mbd
火山引擎开发者社区3 小时前
VeOps 深度调查:让 AI 排障从“能用”走向“可靠”
人工智能
梓豪csd3 小时前
消息订阅中心:谁该收到哪条通知
java·开发语言·人工智能·低代码·自动化·1024程序员节
王中阳Go3 小时前
两个入口答案不一致先别调模型:6 行入口口径对账表(附脚本)
人工智能·后端·python·大语言模型·1024程序员节
武天旭3 小时前
AI安全--AI大模型的隐私保护与法律法规
人工智能·安全
Solange__4 小时前
影视预演怎么用AI生成3D场景?从分镜、Blockout到镜头交付的完整流程
人工智能·游戏·3d·ai·自然语言处理·aigc·游戏策划
冬奇Lab4 小时前
一次 Skill 评测的复盘:我推翻了一份现成的模型选型结论,然后自己造了个评测平台
人工智能·llm·agent
不脱发的程序猿5 小时前
纳指ETF量化策略
数据库·人工智能·1024程序员节