今天调试图像识别模型又搞了一下午,输入图片尺寸问题真是烦死人了 PyTorch的transforms.Resize和PIL.Image.resize参数设置不一样,转来转去数据形状对不上,打印shape看了半天才发现是通道顺序搞反了。还有OpenCV读取的BGR格式和PIL的RGB格式也要注意转换...这种低级错误真的太容易掉坑里了,下次一定要先统一数据预处理流程,不然光是格式转换就能让人崩溃掉 刚才重新跑了一遍代码总算没问题了,accuracy也正常了,松了一口气
登录/注册