MegEngine / MegEngine/Models

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

Offen
#115 0 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen

Dieses Issue hat noch niemand übernommen.

Vorherrschende Sprache
Python
Sterne
306
Forks
96
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

## 使用 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
...

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

Beitragsleitfaden

Für dieses Repository ist kein Beitragsleitfaden indexiert

Erste Schritte

  1. Lies das ganze Issue und danach den Beitragsleitfaden des Projekts.
  2. Schreib ins Issue, dass du es übernimmst — das erspart doppelte Arbeit.
  3. Forke das Repository und arbeite in einem Branch.
  4. Öffne einen Pull Request, der die Issue-Nummer nennt.

Rechercherichtung

Beginne in Models/official/vision/classification/resnet/model.py und reproduziere das Training mit der gezeigten train/val-Verzeichnisstruktur und der Einstellung num_classes=2. Der Issue enthält keine Fehlermeldung und keinen fehlschlagenden Test, daher erfasse zuerst den Trainingsfehler; fertig ist die Aufgabe, wenn das ResNet-Modell erfolgreich auf dem Zwei-Klassen-Datensatz mit Äpfeln und Orangen trainiert wird.

Vom Indexierungsmodell aus dem Issue-Text verfasst.

Bewertung

Tech-Stack
python
Bereich
machine-learning
Issue-Typ
Bug
Schwierigkeit
4/5
Geschätzter Aufwand
3-5 Tage
Aktivitätsstatus
Veraltet
Klarheit
Muss geklärt werden
Anfängerfreundlichkeit
25/100

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.