fix bug in test.py

liangerrr · Jun 8, 2021 · ea73fcd · ea73fcd
1 parent e65dfce
commit ea73fcd
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/notes/演示.pptx b/notes/演示.pptx
diff --git a/src/test.py b/src/test.py
@@ -9,7 +9,7 @@
 image_path = "../imgs/airplane.png"
 image = Image.open(image_path)
 print(image)
-
+image = image.convert('RGB')
 transform = torchvision.transforms.Compose([torchvision.transforms.Resize((32, 32)),
                                             torchvision.transforms.ToTensor()])