报错:
RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton
解决:
img = Image.open(image_path)
改为
img = Image.open(image_path).convert('RGB')。
完成~
报错:
RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton
解决:
img = Image.open(image_path)
改为
img = Image.open(image_path).convert('RGB')。
完成~