diff options
| author | junyanz <junyanzhu89@gmail.com> | 2018-02-09 11:39:35 -0500 |
|---|---|---|
| committer | junyanz <junyanzhu89@gmail.com> | 2018-02-09 11:39:35 -0500 |
| commit | 0ae4f0500e415a6a67689ef9356e8e4779ae5833 (patch) | |
| tree | 25392e96a1b64c8454f7f548886af7dd48aa6bd0 /data/base_dataset.py | |
| parent | 7a5e2cd5f5003e8ca9a0fc3dac14a74b81287881 (diff) | |
code reformatting
Diffstat (limited to 'data/base_dataset.py')
| -rw-r--r-- | data/base_dataset.py | 3 |
1 files changed, 3 insertions, 0 deletions
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): |