深度学习搭建网络模型框架

Posted 2022-12-12 樱木之

整体框架

搭建网络整体结构如下：

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。