这是学习计算机视觉时的练手的一个小项目,非常简单,主要是本人用来入门和熟悉PyTorch框架的
目前已经完成了简单的模型的构建,训练及测试,和将二进制存储的数据集转换为图片的小工具;计划日后添加一些可视化的功能
以后也有实践诸如 AlexNet、VGG、GooLeNet 和 ResNet 等视觉方向常用网络模型的打算
使用的环境:
Python 3.8.5(Anaconda)
PyTorch 1.7.1
CUDA 11.1
测试平台:
Windows 10(RTX3070)
Ubuntu 20.04(RTX3070)
Mac Os Big Sur(CPU)