如题。测试图片太多,都显示出来几百张很恶心,所以保存下来,之后查看也方便。没有任何难度,仅作记录。只需要修改最后一段代码,加一行,再注释一行即可。代码修改如下:
for im_name in im_names:
print '~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~'
print 'Demo for data/demo/{}'.format(im_name)
demo(net, im_name)
plt.savefig("../testfigs/" + im_name)
#plt.show()