added check for unique UUIDs

cvdatasets/_version.py

@@ -1 +1 @@
-__version__ = "0.6.2"
+__version__ = "0.6.3"

cvdatasets/annotations/impl/inat.py

@@ -43,6 +43,10 @@ class BaseINAT_Annotations(BBoxMixin, PartsMixin, BaseAnnotations):
 	def _load_uuids(self):
 		uuid_fnames = [(str(im["id"]), im["file_name"]) for im in self._content["images"]]
 		self.uuids, self.images = map(np.array, zip(*uuid_fnames))
+
+		assert len(np.unique(self.uuids)) == len(self.uuids), \
+			"UUIDs are not unique!"
+
 		self.uuid_to_idx = {uuid: i for i, uuid in enumerate(self.uuids)}
 
 	@property

scripts/display.py

@@ -49,8 +49,8 @@ def main(args):
 	logging.info(f"Loaded {len(data)} {args.subset} images")
 
 	if args.only_class >= 0:
-		logging.info(f"Showing only images from class {args.only_class}")
 		mask = data.labels == args.only_class
+		logging.info(f"Showing only {mask.sum()} images from class {args.only_class}")
 		idxs = np.where(mask)[0]
 	else:
 		start = max(args.start, 0)