From 0ae4f0500e415a6a67689ef9356e8e4779ae5833 Mon Sep 17 00:00:00 2001 From: junyanz Date: Fri, 9 Feb 2018 11:39:35 -0500 Subject: code reformatting --- data/base_dataset.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'data/base_dataset.py') diff --git a/data/base_dataset.py b/data/base_dataset.py index a061a05..7cfac54 100644 --- a/data/base_dataset.py +++ b/data/base_dataset.py @@ -2,6 +2,7 @@ import torch.utils.data as data from PIL import Image import torchvision.transforms as transforms + class BaseDataset(data.Dataset): def __init__(self): super(BaseDataset, self).__init__() @@ -12,6 +13,7 @@ class BaseDataset(data.Dataset): def initialize(self, opt): pass + def get_transform(opt): transform_list = [] if opt.resize_or_crop == 'resize_and_crop': @@ -36,6 +38,7 @@ def get_transform(opt): (0.5, 0.5, 0.5))] return transforms.Compose(transform_list) + def __scale_width(img, target_width): ow, oh = img.size if (ow == target_width): -- cgit v1.2.3-70-g09d2