赞
踩
用ALEXNET训练自己数据集,测试阶段出现的BUG:
一、
错误原因:测试要求用RGB图3通道,而我用的是灰度图。
解决方法:原代码
- # load image
- img_path = "../5.JPG"
- assert os.path.exists(img_path), "file: '{}' dose not exist.".format(img_path)
- img = Image.open(img_path)
-
- plt.imshow(img)
改过后的代码,加一句img = img.covert('RGB')
- # load image
- img_path = "../5.JPG"
- assert os.path.exists(img_path), "file: '{}' dose not exist.".format(img_path)
- img = Image.open(img_path)
- #add
- img = img.convert('RGB')
-
- plt.imshow(img)
完美解决:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。