文件名称：flower-image-classifier:使用预训练的ML模型基于图像训练和预测花朵名称
文件大小：468KB
文件格式：ZIP
更新时间：2024-05-30 16:34:25
HTML
花的图像分类器 使用预训练的ML模型基于图像训练和预测花朵名称 训练新模型 Train.py根据流行的torch.models densitynet121或vgg16创建并训练新模型。 然后，它将保存此新模型的检查点，以便您可以将此检查点用于预测或将来的模型训练。 您可以使用GPU或CPU选项运行该程序（仅供参考：选择CPU选项将使程序花费更长的时间才能完成） 预测花名和概率 Predict.py加载Train.py创建的检查点，并根据花朵的图像执行预测。 然后，它打印出花朵的名称，以及图像在实际中具有预测名称的花朵的概率（或置信度） 示例Jupyter笔记本 还提供了样本Julyter Notebook'Image Classifier.ipynb'。
【文件预览】：
flower-image-classifier-main
----Image Classifier.ipynb(292KB)
----train.py(6KB)
----cat_to_name.json(2KB)
----transform_data.py(2KB)
----predict.py(6KB)
----README.md(842B)
----get_input_args.py(1KB)
----Image Classifier.html(599KB)
----Classifier.py(852B)
----workspace_utils.py(2KB)