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

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(高分辨率版本)
相关推荐
Flynn Liang3 分钟前
LLM 应用安全:Prompt 注入、越权与输入输出护栏
java·人工智能·大模型·langchain4j·llm 安全·prompt 注入·guardrail
吴佳浩4 小时前
《拆解全球最火 AI 开源项目》第一季 · REA
人工智能
AllData公司负责人8 小时前
AI大模型数据开发底座系列说明书,AllData集成Chat2DB开源项目建设AI 数据库分析底座,涵盖AI 洞察/对话SQL/智能视图!
数据库·人工智能·sql·开源·ai 洞察·对话sql
泯泷9 小时前
让 Agent 分清“马上办、等一等、顺手办”
人工智能·llm·agent
鲜于言悠90510 小时前
Claude Fable 5.2 登场,专治复杂推理与长任务
人工智能
联犀11 小时前
如何工程化降低语音交互AI设备的延迟(小智ai,芙崽这样的ai设备)
人工智能
杉岩数据11 小时前
杉岩数据董事长陈坚:从数据存储到行业智能——杉岩的AI时代战略布局
人工智能
冬奇Lab11 小时前
Loop 工程到底是高深的新技术,还是炒冷饭?用训练一个识别狗的模型讲明白
人工智能
冬奇Lab11 小时前
一天一个开源项目(第260篇):Plannotator —— 给 AI 编码 Agent 配一块「可批注的审阅台」,计划和代码改动都能在浏览器里逐条批示
人工智能·开源·claude
泯泷11 小时前
语音,正在成为 AI Agent 最自然的接口
人工智能·llm·agent