|
|
@@ -27,9 +27,14 @@ class _base_mixin(abc.ABC):
|
|
|
|
|
|
def get_example(self, i):
|
|
|
im, lab = super(_base_mixin, self).get_example(i)
|
|
|
+
|
|
|
if isinstance(im, list):
|
|
|
im = np.array(im)
|
|
|
- return im * 2 - 1, lab
|
|
|
+
|
|
|
+ if np.logical_and(0 <= im, im <= 1).all():
|
|
|
+ im = im * 2 -1
|
|
|
+
|
|
|
+ return im, lab
|
|
|
|
|
|
|
|
|
class BaseDataset(_base_mixin,
|
