Browse Source

some fix in image wrapper init method

Dimitri Korsch 5 years ago
parent
commit
4b0cd12bc5
2 changed files with 4 additions and 2 deletions
  1. 1 1
      cvdatasets/_version.py
  2. 3 1
      cvdatasets/dataset/image.py

+ 1 - 1
cvdatasets/_version.py

@@ -1 +1 @@
-__version__ = "0.5.2"
+__version__ = "0.5.3"

+ 3 - 1
cvdatasets/dataset/image.py

@@ -25,9 +25,11 @@ class ImageWrapper(object):
 	def __init__(self, im_path, label, parts=None, mode="RGB", part_rescale_size=None, center_cropped=True):
 
 		self.mode = mode
-		self.im = im_path
+		self._im = None
 		self._im_array = None
 
+		self.im = im_path
+
 		self.label = label
 		self.parts = Parts(self.im, parts, part_rescale_size, center_cropped)