From 0ae4f0500e415a6a67689ef9356e8e4779ae5833 Mon Sep 17 00:00:00 2001 From: junyanz Date: Fri, 9 Feb 2018 11:39:35 -0500 Subject: code reformatting --- util/image_pool.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'util/image_pool.py') diff --git a/util/image_pool.py b/util/image_pool.py index ada1627..634fd81 100644 --- a/util/image_pool.py +++ b/util/image_pool.py @@ -1,5 +1,4 @@ import random -import numpy as np import torch from torch.autograd import Variable @@ -24,7 +23,7 @@ class ImagePool(): else: p = random.uniform(0, 1) if p > 0.5: - random_id = random.randint(0, self.pool_size-1) + random_id = random.randint(0, self.pool_size - 1) tmp = self.images[random_id].clone() self.images[random_id] = image return_images.append(tmp) -- cgit v1.2.3-70-g09d2