Hacktoberfest 2026:维护者为十月标记出来的 issue,仍然开放、适合新手。 浏览 Hacktoberfest issue

[T5] EfficientNet-B0: load ImageNet-pretrained model with a 3-class head

已关闭 适合新手
#71 0 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看

维护者通常 1 天内回复

还没有人认领这个 Issue。

评估

难度
2/5
预计耗时
1-3 小时
新手友好度
82/100
Issue 类型
功能
描述清晰度
描述清楚
活跃度
活跃
技术栈
python, tensorflow

调研方向

从 SOTA_EfficientNetB0.ipynb 开始,查看 EfficientNetB0 加载和数据加载示例;issue 指定了模型设置、head、共享数据划分和预处理。构建模型,并从 animal/train 运行一个 batch;当预测的形状为 (batch, 3),且总参数量和可训练参数量均已打印时,即完成。

由索引模型根据 Issue 内容生成。

描述

task-5 ws4-efficientnet

Assignment task 5 (SOTA fine-tuning).

File: SOTA_EfficientNetB0.ipynb

Tasks
  • tf.keras.applications.EfficientNetB0(include_top=False, weights="imagenet", input_shape=(64, 64, 3))
  • Add head: GlobalAveragePooling → Dropout → Dense(3, softmax)
  • Correct preprocessing: EfficientNet includes its own rescaling layer – feed raw 0–255 pixels, do not divide by 255
  • Print total and trainable parameter counts

Use the shared split in animal/train, animal/val, animal/test (64×64 RGB PNG, 700/150/150 images per class). Do not re-split or resize. No data augmentation (group decision).

Done when

Model builds and predicts shape (batch, 3) on one batch from animal/train.

Depends on

#49


Branch: feat/71-efficientnet-build · open a PR into main with Closes #71 in the description.

主要语言
Jupyter Notebook
星标
0
派生
0
平均合并
2 小时 30 分钟
30 天内合并 PR
23

环境准备

这个项目没有提供开发容器、Dockerfile 或贡献指南,环境需要你自己搭建:先看它的 README,通用步骤见我们的新手贡献指南。

从这里开始

  1. 先读完整个 Issue,再读项目的贡献指南。
  2. 在 Issue 下留言说明你要接手 —— 这能避免两个人做同样的事。
  3. Fork 仓库,在一个分支上完成修改。
  4. 提交 Pull Request,并在描述里引用这个 Issue 编号。

相似的 Issue

更多 Machine Learning Issue

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。