[dataloader]会以一个很奇怪的顺序来读取图片,如果我们
2018-12-14 本文已影响10人
VanJordan
from PIL import Image
drain=os.path.join(self.test_data,'%d.jpg'%i)
img = Image.open(drain).convert('RGB')
img = self.data_transform['test'](img)
img=img.to(device)