diff --git a/test/assets/results/style_transfer/cat_candy.gt b/test/assets/results/style_transfer/cat_candy.gt
new file mode 100644
index 0000000..c2afd55
Binary files /dev/null and b/test/assets/results/style_transfer/cat_candy.gt differ
diff --git a/test/assets/results/style_transfer/cat_candy_fast.gt b/test/assets/results/style_transfer/cat_candy_fast.gt
new file mode 100644
index 0000000..6182c9c
Binary files /dev/null and b/test/assets/results/style_transfer/cat_candy_fast.gt differ
diff --git a/test/assets/results/style_transfer/cat_mosaic.gt b/test/assets/results/style_transfer/cat_mosaic.gt
new file mode 100644
index 0000000..f56190e
Binary files /dev/null and b/test/assets/results/style_transfer/cat_mosaic.gt differ
diff --git a/test/assets/results/style_transfer/cat_mosaic_fast.gt b/test/assets/results/style_transfer/cat_mosaic_fast.gt
new file mode 100644
index 0000000..9d951d3
Binary files /dev/null and b/test/assets/results/style_transfer/cat_mosaic_fast.gt differ
diff --git a/test/assets/results/style_transfer/cat_princess.gt b/test/assets/results/style_transfer/cat_princess.gt
new file mode 100644
index 0000000..dc36a3c
Binary files /dev/null and b/test/assets/results/style_transfer/cat_princess.gt differ
diff --git a/test/assets/results/style_transfer/cat_princess_fast.gt b/test/assets/results/style_transfer/cat_princess_fast.gt
new file mode 100644
index 0000000..10e0938
Binary files /dev/null and b/test/assets/results/style_transfer/cat_princess_fast.gt differ
diff --git a/test/assets/results/style_transfer/cat_udnie.gt b/test/assets/results/style_transfer/cat_udnie.gt
new file mode 100644
index 0000000..23a300c
Binary files /dev/null and b/test/assets/results/style_transfer/cat_udnie.gt differ
diff --git a/test/assets/results/style_transfer/cat_udnie_fast.gt b/test/assets/results/style_transfer/cat_udnie_fast.gt
new file mode 100644
index 0000000..be67867
Binary files /dev/null and b/test/assets/results/style_transfer/cat_udnie_fast.gt differ
diff --git a/test/ex_vision/style_transfer/candy_test.exs b/test/ex_vision/style_transfer/candy_test.exs
index 8c570d7..4766dc1 100644
--- a/test/ex_vision/style_transfer/candy_test.exs
+++ b/test/ex_vision/style_transfer/candy_test.exs
@@ -1,14 +1,14 @@
 defmodule Configuration do
   def configuration do
     %{
-      ExVision.StyleTransfer.CandyTest  => [module: ExVision.StyleTransfer.Candy, gt_file: "cat_candy.out"],
-      ExVision.StyleTransfer.CandyFastTest  => [module: ExVision.StyleTransfer.CandyFast, gt_file: "cat_candy_fast.out"],
-      ExVision.StyleTransfer.PrincessTest  => [module: ExVision.StyleTransfer.Princess, gt_file: "cat_princess.out"],
-      ExVision.StyleTransfer.PrincessFastTest  => [module: ExVision.StyleTransfer.PrincessFast, gt_file: "cat_princess_fast.out"],
-      ExVision.StyleTransfer.UdnieTest  => [module: ExVision.StyleTransfer.Udnie, gt_file: "cat_udnie.out"],
-      ExVision.StyleTransfer.UdnieFastTest  => [module: ExVision.StyleTransfer.UdnieFast, gt_file: "cat_udnie_fast.out"],
-      ExVision.StyleTransfer.MosaicTest  => [module: ExVision.StyleTransfer.Mosaic, gt_file: "cat_mosaic.out"],
-      ExVision.StyleTransfer.MosaicFastTest  => [module: ExVision.StyleTransfer.MosaicFast, gt_file: "cat_mosaic_fast.out"],
+      ExVision.StyleTransfer.CandyTest  => [module: ExVision.StyleTransfer.Candy, gt_file: "cat_candy.gt"],
+      ExVision.StyleTransfer.CandyFastTest  => [module: ExVision.StyleTransfer.CandyFast, gt_file: "cat_candy_fast.gt"],
+      ExVision.StyleTransfer.PrincessTest  => [module: ExVision.StyleTransfer.Princess, gt_file: "cat_princess.gt"],
+      ExVision.StyleTransfer.PrincessFastTest  => [module: ExVision.StyleTransfer.PrincessFast, gt_file: "cat_princess_fast.gt"],
+      ExVision.StyleTransfer.UdnieTest  => [module: ExVision.StyleTransfer.Udnie, gt_file: "cat_udnie.gt"],
+      ExVision.StyleTransfer.UdnieFastTest  => [module: ExVision.StyleTransfer.UdnieFast, gt_file: "cat_udnie_fast.gt"],
+      ExVision.StyleTransfer.MosaicTest  => [module: ExVision.StyleTransfer.Mosaic, gt_file: "cat_mosaic.gt"],
+      ExVision.StyleTransfer.MosaicFastTest  => [module: ExVision.StyleTransfer.MosaicFast, gt_file: "cat_mosaic_fast.gt"],
     }
   end
 end