1.2MBZIP
[Beschreibung der Ressourcen]
基于HAM10000数据集实现皮肤癌分类python源码+使用说明.zip
这是一个用于训练图像分类模型的代码。在运行代码之前,用户需要安装以下依赖库:argparse, os, pandas, numpy, PIL, datasets, torchvision, tqdm和transformers。用户还需要从Hugging Face上下载所需的预训练模型。
参数说明
– `–metadata_path`:metadata文件的路径。默认为”./archive/HAM10000_metadata.csv”。
– `–images_dir`:图像文件夹的路径。默认为”./archive/HAM10000_images/”。
– `–model_dir`:预训练模型的路径。默认为”../model/vit-large-patch16-224-in21k”。
– `–checkpoints_dir`:保存检查点文件的文件夹路径。默认为”./checkpoints”。
– `–learning_rate`:学习率。默认为1e-5。
– `–batch_size`:批大小。默认为64。
– `–epochs`:训练轮数。默认为5。
– `–warmup_ratio`:预热步骤的比例。默认为0.1。
– `–split`:训练-验证数据集的分割比例。默认为0.8。
– `–gpu`:指定使用哪张GPU。默认为”0″。
– `–logging_steps`:每隔多少步记录一次训练日志。默认为50。
用户可以在命令行中传递这些参数,例如:
“`shell
python train-hf.py –metadata_path ./archive/HAM4000_metadata.csv \
–images_dir ./archive/HAM10000_images/ \
–checkpoints_dir ./checkpoints \
–learning_rate 1e-4 \
–batch_size 64 \
–epochs 20 \
–warmup_ratio 0.1 \
–model_dir ../model/vit-large-patch16-224-in21k \
–gpu 5,6,7 \
–logging_steps 1
“`
在代码运行过程中,会执行以下步骤:
1. 读取metadata文件,获取图像文件名和标签。
2. 将图像读入内存,并随机打乱。
3. 将数据集划分为训练集和验证集。
4. 对图像进行预处理,包括随机裁剪、归一化和转换为tensor。
5. 加载预训练模型,构建分类器。
6. 训练模型,并在验证集上评估模型性能。
7. 在训练过程中,每隔logging_steps步记录一次训练日志,包括损失值、准确率等指标。
8. 在训练结束后,保存模型的权重文件到checkpoints_dir文件夹中。
[Bemerkungen]
该项目是个人毕设/课设/大作业项目,代码都经过本地调试测试,功能ok才上传,高分作品,可快速上手运行!欢迎下载使用,可用于小白学习、进阶。
Diese Ressource ist vor allem für Studenten, Lehrer oder Praktiker der Computer-, Kommunikations-, künstliche Intelligenz, Automatisierung und anderen verwandten Majors zum Herunterladen und Verwenden, und kann auch als die endgültige Kurs-Design, Kursarbeit, Graduierung Design verwendet werden.
Das Projekt als Ganzes hat einen hohen Lernwert! Starke Grundfähigkeiten können auf dieser Basis verändert und angepasst werden, um verschiedene Funktionen zu erreichen.
欢迎下载使用,也欢迎交流学习!
Ressourcenerklärung (der Kauf gilt als Zustimmung zu dieser Erklärung): 1. jede Operation in der Website-Plattform gilt als gelesen zu haben und stimmte der Unterseite der Website-Registrierung Vereinbarung und Haftungsausschluss, diese Website Ressourcen wurden ultra-niedrigen Preis, und bietet keine technische Unterstützung (2) Einige Netzwerk-Benutzer teilen sich die Netto-Disc-Adresse kann ungültig sein, wie das Auftreten von Fehlern, senden Sie bitte eine E-Mail an den Kundendienst code711cn#qq.com (ersetzen Sie # mit @) wird bis zu senden gemacht werden 3. diese Website bietet alle herunterladbaren Ressourcen (Software, etc.) Website, um sicherzustellen, dass keine negativen Veränderungen; aber diese Website kann nicht garantieren, die Richtigkeit, Sicherheit und Integrität der Ressourcen, die Benutzer-Downloads nach eigenem Ermessen, kommunizieren wir für die Zwecke der nicht alle den Quellcode zu lernen ist nicht 100% fehlerfrei oder keine Bugs; Sie müssen eine bestimmte Grundlage zu können, lesen und verstehen den Code, um in der Lage, das Debugging zu ändern haben! Code zu ändern und Fehler zu beheben. Gleichzeitig müssen die Nutzer dieser Seite verstehen, dass der Source Code Convenience Store keine Rechte an der zum Download bereitgestellten Software besitzt, das Urheberrecht gehört dem rechtlichen Eigentümer der Ressource. 4. alle Ressourcen auf dieser Website nur für Lern- und Forschungszwecke, bitte müssen innerhalb von 24 Stunden der heruntergeladenen Ressourcen gelöscht werden, verwenden Sie nicht für kommerzielle Zwecke, da sonst die rechtlichen Streitigkeiten, die sich aus der Website und der Herausgeber der Sicherheiten Haftung Website und wird nicht getragen werden! (5) Aufgrund der Reproduzierbarkeit der Ressourcen sind einmal erworbene Beträge nicht erstattungsfähig, und auch das Guthaben wird nicht zurückerstattet.