Spaces:
Runtime error
Runtime error
artelabsuper
commited on
Commit
•
ad62063
1
Parent(s):
459c031
remain scaling commented on test script
Browse files
test.py
CHANGED
|
@@ -10,6 +10,7 @@ from models.modelNetC import Generator as GC
|
|
| 10 |
|
| 11 |
|
| 12 |
|
|
|
|
| 13 |
DEVICE='cuda'
|
| 14 |
model_type = 'model_c'
|
| 15 |
|
|
@@ -35,7 +36,7 @@ generator.eval()
|
|
| 35 |
|
| 36 |
preprocess = transforms.Compose([
|
| 37 |
transforms.Grayscale(),
|
| 38 |
-
# transforms.Resize((
|
| 39 |
transforms.ToTensor()
|
| 40 |
])
|
| 41 |
input_img = Image.open('demo_imgs/fake.jpg')
|
| 10 |
|
| 11 |
|
| 12 |
|
| 13 |
+
# DEVICE='cpu'
|
| 14 |
DEVICE='cuda'
|
| 15 |
model_type = 'model_c'
|
| 16 |
|
| 36 |
|
| 37 |
preprocess = transforms.Compose([
|
| 38 |
transforms.Grayscale(),
|
| 39 |
+
# transforms.Resize((128, 128)),
|
| 40 |
transforms.ToTensor()
|
| 41 |
])
|
| 42 |
input_img = Image.open('demo_imgs/fake.jpg')
|