学习笔记(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,是单通道的图像

发布了46 篇原创文章 · 获赞 5 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/qq_36808245/article/details/104196545
今日推荐