resnet分类模型无法指定分类数量
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
- Leggi tutta la issue e poi la guida ai contributi del progetto.
- Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
- Fai un fork del repository e lavora su un branch.
- 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