array 读取图片 tensor 模型预测
一般过程:读取数据np->tensor->model->result->np->画图
shape确保图像输入输出尺寸正确
读取图片
将在GPU上运行的tensor类型转变成在CPU上运行的np类型
三类计算机视觉任务的输入:
分类:输入[8,3,823,1623] 8为批次 3为通道数 [823,1623]图像大小
3分类 ,输出[8,3]
分割:输入[8,3,823,1623] 输出[8,823,1623]
检测:输入{[8,3,823,1623],[boxes,label]}
输出{scores,label,boxes}