MegEngine / MegEngine/Models

resnet分类模型无法指定分类数量

Aperta
#115 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub

Nessuno ha ancora preso questa issue.

Lingua principale
Python
Stelle
306
Fork
96
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

## 使用 classification resnet进行分类苹果/橘子训练时,指定分类数量为2时,无法训练
Models/official/vision/classification/resnet/model.py
resnet50只支持1000分类数据集训练,我只创建2分类数据时,修改num_classes=2时,无法正常完成训练

## 任务描述
数据集格式是这样的:
/path/to/imagenet
train
apple
xxx.jpg
...
orangle
xxx.jpg
...
...
val
apple
xxx.jpg
...
orangle
xxx.jpg
...

## 目标
由于我时初学,在经过自己各种修改后,仍然无法解决,需要帮忙。谢谢

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Direzione di ricerca

Inizia da Models/official/vision/classification/resnet/model.py e riproduci l’addestramento con la struttura delle directory train/val mostrata e l’impostazione num_classes=2. L’issue non include un messaggio di errore né un test fallito, quindi cattura prima il problema dell’addestramento; il lavoro è completato quando il modello ResNet viene addestrato correttamente sul dataset a due classi di mele e arance.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Valutazione

Stack tecnologico
python
Ambito
machine-learning
Tipo di issue
Bug
Difficoltà
4/5
Tempo stimato
3-5 giorni
Stato di attività
Ferma
Chiarezza
Da chiarire
Idoneità per principianti
25/100

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.