some minor changes

cvdatasets/_version.py

@@ -1 +1 @@
-__version__ = "0.9.2"
+__version__ = "0.11.0"

cvdatasets/annotation/base.py

@@ -109,6 +109,14 @@ class BaseAnnotations(abc.ABC):
 	def __getitem__(self, uuid) -> Tuple[str, int]:
 		return self.image(uuid), self.label(uuid)
 
+	@property
+	def labels(self):
+		return self._labels
+
+	@labels.setter
+	def labels(self, labels):
+		self._orig_labels, self._labels = np.unique(labels, return_inverse=True)
+
 	def image_path(self, image) -> str:
 		return str(self.root / self.images_folder / image)
 

cvdatasets/resize.py

@@ -19,6 +19,8 @@ def resize(name, *, source: Path, dest: Path, size: int, fit_short: bool = False
 	dst = dest / name
 
 	im  = read_image(src)
+	if im.mode == "RGBA":
+		im = im.convert("RGB")
 	w, h = im.size
 
 	if fit_short:

cvdatasets/utils/image.py

@@ -6,7 +6,7 @@ from PIL import Image
 from cvdatasets.utils import retry_operation
 
 def read_image(im_path, n_retries=5):
-	assert isfile(im_path), "Image \"{}\" does not exist!".format(im_path)
+	#assert isfile(im_path), f"Image \"{im_path}\" does not exist!"
 	return retry_operation(n_retries, Image.open, im_path, mode="r")
 
 

requirements.txt

@@ -4,4 +4,4 @@ pillow~=9.0
 matplotlib
 pyyaml
 
-cvargparse~=0.3
+cvargparse~=0.5