学习笔记(01):深度学习之图像识别 核心技术与案例实战-模型训练和测试

立即学习:https://edu.csdn.net/course/play/26257/326455?utm_source=blogtoedu

 testsize预测网络输入大小

net初始化网络模型权重

roi储存了数据，但是是rgb格式的，注意训练时是bgr顺序

transformer是预处理接口，完成减去均值mean和通道交换transpose，caffe 输入格式是四维张量:通道、图像的宽高 以及batchsize

out调用forward接口可以得到图像输出

res最后一层输出，包含三个通道（三分类），使用argmax取最大概率的标签reslabel，是单通道的图像