文件名称:image-classifier-aipnd:使用Python Nanodegree(Udacity)进行AI编程的图像分类器项目
文件大小:244KB
文件格式:ZIP
更新时间:2024-06-06 14:05:57
JupyterNotebook
图像分类器 使用Python Nanodegree(Udacity)进行AI编程的图像分类器项目 示例命令 python train.py --arch densenet --hidden_units 1000,500 --epochs 10 --learning_rate 0.001 --gpu true --data_dir flowers/ --save_dir checkpoint.pth python predict.py --checkpoint checkpoint.pth --input flowers/test/32/image_051001.jpg --top_k 5 --category_names cat_to_name.json --gpu true 训练 要训练模型, train.py使用所需的模型架构(densenet或vgg)并运行image文件夹的
【文件预览】:
image-classifier-aipnd-master
----cat_to_name.json(2KB)
----.gitignore(1KB)
----predict.py(2KB)
----README.md(3KB)
----command_utils.py(3KB)
----train.py(3KB)
----model_utils.py(1KB)
----data_loading.py(2KB)
----notebooks()
--------Image Classifier Project.ipynb(330KB)