changing indents bruh
Browse files- test_api.py +2 -2
test_api.py
CHANGED
|
@@ -58,10 +58,10 @@ def run_adain(content_dir, style_dataset_pth, out_dir, alpha=1.0, vgg_pth='vgg_n
|
|
| 58 |
# Timer
|
| 59 |
times = []
|
| 60 |
|
| 61 |
-
|
| 62 |
# do i need to stick a dataloader around this? idk
|
| 63 |
|
| 64 |
-
|
| 65 |
style_img = style_item['image']
|
| 66 |
print(style_img)
|
| 67 |
style_tensor = t(style_img).unsqueeze(0).to(device)
|
|
|
|
| 58 |
# Timer
|
| 59 |
times = []
|
| 60 |
|
| 61 |
+
style_ds = load_dataset(style_dataset_pth, split="train")
|
| 62 |
# do i need to stick a dataloader around this? idk
|
| 63 |
|
| 64 |
+
for style_idx, style_item in enumerate(style_ds):
|
| 65 |
style_img = style_item['image']
|
| 66 |
print(style_img)
|
| 67 |
style_tensor = t(style_img).unsqueeze(0).to(device)
|