文件名称:利用Pytorch搭建简单的图像分类模型(完整模型,从读取数据到最后训练验证)
文件大小:649.61MB
文件格式:ZIP
更新时间:2022-09-27 21:08:09
pytorch resnet 分类算法
这是一个完整的基于Pytorch 的分类模型,下载以后根据博客中讲的读取数据、搭建模型、训练验证等可以自行修改代码,以符合自身需求。整个项目都有比较清晰的注释,比较适合新手上手学习。
【文件预览】:
classification_project
----data_utlis.py(2KB)
----logs()
----data()
--------CIFAR10()
----test.py(2KB)
----train.py(3KB)
----__pycache__()
--------base_model.cpython-38.pyc(1KB)
--------train_test_utils.cpython-38.pyc(2KB)
--------model.cpython-38.pyc(3KB)
--------data_utlis.cpython-38.pyc(1KB)
----model.py(5KB)
----weights()
----train_test_utils.py(5KB)
----readme.txt(318B)