review_changes

Author: sanjaychelliah

clarifai_datautils/image/annotation_conversion/utils.py

@@ -22,8 +22,6 @@ def __init__(
     Args:
         path (str): The path to the clarifai dataset.
 
-    Returns:
-        A Datumaro dataset object.
     """
     self.main_path = main_path
     self.image_list = os.listdir(os.path.join(self.main_path, 'inputs'))
@@ -46,13 +44,10 @@ def convert(self) -> Dataset:
     return dataset
 
   def create_item(self, image_path: str) -> DatasetItem:
-    """Creates a Datumaro item from an image path.
-    """
-    #read the image
+    """Creates a Datumaro item from an image path."""
     image_full_path = os.path.join(self.main_path, 'inputs', image_path)
     image_data = Image.from_file(image_full_path)
     width, height = PIL.Image.open(image_full_path).size
-    # read the json file
     try:
       with open(
           os.path.join(self.main_path, 'annotations', image_path.split('.png')[0] + '.json'),
@@ -70,7 +65,6 @@ def create_item(self, image_path: str) -> DatasetItem:
           self.label_map[label] = value
           annotations.append(Bbox(x=x, y=y, w=w, h=h, label=value))
 
-    #file not found error
     except FileNotFoundError:
       annotations = []
 
@@ -86,23 +80,19 @@ def clarifai_bbox_to_datumaro_bbox(self, clarifai_bbox, width, height) -> Tuple[
     return obj_box
 
   def check_folder(self):
-    """Checks the clarifai folder format.
-    """
+    """Checks the clarifai folder format."""
     if not os.path.exists(self.main_path):
       raise AnnotationsDatasetError(f'Folder not found at {self.main_path}')
 
-    #check if the folder has two subfolders named 'inputs' and 'annotations'
     if not os.path.exists(os.path.join(self.main_path, 'inputs')):
       raise AnnotationsFormatError(
           f'Folder does not contain an "inputs" folder at {self.main_path}')
     if not os.path.exists(os.path.join(self.main_path, 'annotations')):
       raise AnnotationsFormatError(
           f'Folder does not contain an "annotations" folder at {self.main_path}')
 
-    #check if the 'inputs' folder contains images
     if not all(img.endswith('.png') for img in self.image_list):
       raise AnnotationsFormatError(f'Folder should only contain images at {self.main_path}/inputs')
-    #check if the 'annotations' folder contains json files
     if not all(img.endswith('.json') for img in self.annotations_list):
       raise AnnotationsFormatError(
           f'Folder should only contain annotations at {self.main_path}/annotations')