整体框架
搭建网络整体结构如下:
AlexNet
|---envrioment (option)
|---requirement.txt (option)
|---legacy (option)
|---model2.py (option)
|---utils (option)
|---resize_img.py (option)
|---convertion.py (option)
|---preprocess.py (option)
|---dataset.py
|---model.py
|---train.py
|---inference.py
|---README.md
其中必须要书写model.py
,dataset.py
,train.py
,inference.py
。
model.py
该模块用于搭建网络模型,可以参考如下链接https://blog.csdn.net/qq_43369406/article/details/127932585
。
dataset.py
该模块用于读取数据并生成dataset对象传入train脚本中,可以参考如下链接https://blog.csdn.net/qq_43369406/article/details/127932597
。
train.py
该模块较为复杂,套娃。用于训练模型并生成.model/.pkl/.pth/.h5
等模型权重文件。可以参考链接https://blog.csdn.net/qq_43369406/article/details/127932640
。
inference.py
该模块用于预测和推理,可以参考链接https://blog.csdn.net/qq_43369406/article/details/127932629
。