From c30655f3bc88a5ec50f3db8f794442f7a76419e1 Mon Sep 17 00:00:00 2001 From: yhkwon-DT01 Date: Thu, 2 Sep 2021 17:30:18 +0900 Subject: [PATCH] fix transforms for val an test process --- examples/cats_and_dogs.ipynb | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/examples/cats_and_dogs.ipynb b/examples/cats_and_dogs.ipynb index f5cbfab..54febb7 100644 --- a/examples/cats_and_dogs.ipynb +++ b/examples/cats_and_dogs.ipynb @@ -364,9 +364,8 @@ "\n", "val_transforms = transforms.Compose(\n", " [\n", - " transforms.Resize((224, 224)),\n", - " transforms.RandomResizedCrop(224),\n", - " transforms.RandomHorizontalFlip(),\n", + " transforms.Resize(256),\n", + " transforms.CenterCrop(224),\n", " transforms.ToTensor(),\n", " ]\n", ")\n", @@ -374,9 +373,8 @@ "\n", "test_transforms = transforms.Compose(\n", " [\n", - " transforms.Resize((224, 224)),\n", - " transforms.RandomResizedCrop(224),\n", - " transforms.RandomHorizontalFlip(),\n", + " transforms.Resize(256),\n", + " transforms.CenterCrop(224),\n", " transforms.ToTensor(),\n", " ]\n", ")\n" @@ -6250,4 +6248,4 @@ }, "nbformat": 4, "nbformat_minor": 1 -} +} \ No newline at end of file