update output location

test_api.py

@@ -76,26 +76,11 @@ def run_adain(content_dir, style_dataset_pth, out_dir, alpha=1.0, dataset_size=1
 				style_img = style_img.convert("RGB")
 			style_tensor = t(style_img).unsqueeze(0).to(device)
     
-			# Start time
-			tic = time.perf_counter()
-            
 			# Execute style transfer
 			with torch.no_grad():
 				out_tensor = style_transfer(content_tensor, style_tensor, model.encoder, model.decoder, alpha).cpu()
-        
-			# End time
-			toc = time.perf_counter()
-			print("Content: " + content_pth.stem + ". Style: " \
-				+ str(idx) + '. Alpha: ' + str(alpha) + '. Style Transfer time: %.4f seconds' % (toc-tic))
-			times.append(toc-tic)
             
 			# Save image
-			out_pth = out_dir + content_pth.stem + '_style_' + str(idx) + '_alpha' + str(alpha)
-			out_pth += content_pth.suffix
+			out_pth = os.path.join(out_dir, content_pth.stem + '_style_' + str(idx) + '_alpha' + str(alpha) + content_pth.suffix)
 			save_image(out_tensor, out_pth)
-
-	# Remove runtime of first iteration because it is flawed for some unknown reason
-	if len(times) > 1:
-		times.pop(0)
-		avg = sum(times)/len(times)
-		print("Average style transfer time: %.4f seconds" % (avg))
+			print(f"Style transferred image saved to {out_pth}")