diff options
Diffstat (limited to 'util/image_pool.py')
| -rw-r--r-- | util/image_pool.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/util/image_pool.py b/util/image_pool.py index 9f34a09..5a242e6 100644 --- a/util/image_pool.py +++ b/util/image_pool.py @@ -2,6 +2,8 @@ import random import numpy as np import torch from torch.autograd import Variable + + class ImagePool(): def __init__(self, pool_size): self.pool_size = pool_size |