|
|
@@ -6,10 +6,11 @@ from cvdatasets.dataset.mixins.base import BaseMixin
|
|
|
|
|
|
class TransformMixin(BaseMixin):
|
|
|
|
|
|
- def __init__(self, size, *args, **kwargs):
|
|
|
+ def __init__(self, size, part_size=None, *args, **kwargs):
|
|
|
super(TransformMixin, self).__init__(*args, **kwargs)
|
|
|
|
|
|
self.size = size
|
|
|
+ self.part_size = size if part_size is None else part_size
|
|
|
|
|
|
@abc.abstractmethod
|
|
|
def transform(self, im_obj):
|
|
|
@@ -29,3 +30,14 @@ class TransformMixin(BaseMixin):
|
|
|
@size.setter
|
|
|
def size(self, value):
|
|
|
self._size = Size(value)
|
|
|
+
|
|
|
+ @property
|
|
|
+ def part_size(self):
|
|
|
+ if chainer.config.train:
|
|
|
+ return self._part_size // 0.875
|
|
|
+ else:
|
|
|
+ return self._part_size
|
|
|
+
|
|
|
+ @part_size.setter
|
|
|
+ def part_size(self, value):
|
|
|
+ self._part_size = Size(value)
|
