本文包含瓷砖数据集,为四分类任务,首先运行train_alecnet.py进行模型训练,训练完成后运行gui.py进行见面展示,选择图片进行模型调用判断所选择的图片。 运行环境:python:3.7 torch:1.13
bubblesli/-alexnet_cz_cv
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
| Name | Name | Last commit date | ||
|---|---|---|---|---|
本文包含瓷砖数据集,为四分类任务,首先运行train_alecnet.py进行模型训练,训练完成后运行gui.py进行见面展示,选择图片进行模型调用判断所选择的图片。 运行环境:python:3.7 torch:1.13