Update test_api.py
Browse files- test_api.py +1 -1
test_api.py
CHANGED
|
@@ -86,7 +86,7 @@ def run_adain(content_dir, style_dataset_pth, out_dir, alpha=1.0, dataset_size=1
|
|
| 86 |
# End time
|
| 87 |
toc = time.perf_counter()
|
| 88 |
print("Content: " + content_pth.stem + ". Style: " \
|
| 89 |
-
+ str(
|
| 90 |
times.append(toc-tic)
|
| 91 |
|
| 92 |
# Save image
|
|
|
|
| 86 |
# End time
|
| 87 |
toc = time.perf_counter()
|
| 88 |
print("Content: " + content_pth.stem + ". Style: " \
|
| 89 |
+
+ str(idx) + '. Alpha: ' + str(alpha) + '. Style Transfer time: %.4f seconds' % (toc-tic))
|
| 90 |
times.append(toc-tic)
|
| 91 |
|
| 92 |
# Save image
|