update for pep8

midv500/convert_dataset.py

@@ -6,6 +6,7 @@
 from tqdm import tqdm
 from midv500.utils import list_annotation_paths_recursively, get_bbox_inside_image, create_dir
 
+
 def convert(root_dir: str, export_dir: str):
     """
     Walks inside root_dir (should oly contain original midv500 dataset folders),
@@ -31,8 +32,8 @@ def convert(root_dir: str, export_dir: str):
     print("Converting to coco.")
     for ind, rel_annotation_path in enumerate(tqdm(annotation_paths)):
         # get image path
-        rel_image_path = rel_annotation_path.replace("ground_truth","images")
-        rel_image_path = rel_image_path.replace("json","tif")
+        rel_image_path = rel_annotation_path.replace("ground_truth", "images")
+        rel_image_path = rel_image_path.replace("json", "tif")
 
         # load image
         abs_image_path = os.path.join(root_dir, rel_image_path)
@@ -63,9 +64,9 @@ def convert(root_dir: str, export_dir: str):
         # create mask from poly coords
         mask = np.zeros(image.shape, dtype=np.uint8)
         mask_coords_np = np.array(mask_coords, dtype=np.int32)
-        cv2.fillPoly(mask, mask_coords_np.reshape(-1, 4, 2), color=(255,255,255))
+        cv2.fillPoly(mask, mask_coords_np.reshape(-1, 4, 2), color=(255, 255, 255))
         mask = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY)
-        mask = cv2.threshold(mask, 0,255, cv2.THRESH_BINARY)[1]
+        mask = cv2.threshold(mask, 0, 255, cv2.THRESH_BINARY)[1]
 
         # get voc style bounding box coordinates [minx, miny, maxx, maxy] of the mask
         label_xmin = min([pos[0] for pos in mask_coords])
@@ -83,7 +84,7 @@ def convert(root_dir: str, export_dir: str):
         annotation_dict = dict()
         annotation_dict["iscrowd"] = 0
         annotation_dict["image_id"] = image_dict['id']
-        annotation_dict['category_id'] = 1 # id card
+        annotation_dict['category_id'] = 1  # id card
         annotation_dict['ignore'] = 0
         annotation_dict['id'] = ind
 

midv500/download_dataset.py

@@ -53,6 +53,7 @@
              'ftp://smartengines.com/midv-500/dataset/49_usa_ssn82.zip',
              'ftp://smartengines.com/midv-500/dataset/50_xpo_id.zip']
 
+
 def download_dataset(download_dir: str):
     """
     This script downloads the MIDV-500 dataset and unzips the folders.
@@ -66,18 +67,19 @@ def download_dataset(download_dir: str):
         print('Downloaded:', link[40:])
         # unzip zip file
         print('Unzipping:', link[40:])
-        zip_path = os.path.join(download_dir,link[40:])
+        zip_path = os.path.join(download_dir, link[40:])
         unzip(zip_path, download_dir)
         print('Unzipped:', link[40:].replace('.zip', ''))
         # remove zip file
         os.remove(zip_path)
 
+
 if __name__ == '__main__':
     # construct the argument parser
     ap = argparse.ArgumentParser()
 
     # add the arguments to the parser
-    ap.add_argument("download_dir", default = "data/", help="Directory for MIDV-500 dataset to be downloaded.")
+    ap.add_argument("download_dir", default="data/", help="Directory for MIDV-500 dataset to be downloaded.")
     args = vars(ap.parse_args())
 
     # download dataset

midv500/utils.py

@@ -87,7 +87,7 @@ def list_annotation_paths_recursively(directory: str, ignore_background_only_one
                         continue
 
                 relative_filepath = abs_filepath.split(directory)[-1]
-                relative_filepath = relative_filepath.replace("\\", "/") # for windows
+                relative_filepath = relative_filepath.replace("\\", "/")  # for windows
                 relative_filepath_list.append(relative_filepath)
 
     number_of_files = len(relative_filepath_list)

setup.cfg

@@ -1,7 +1,7 @@
 [flake8]
 max-line-length = 119
 exclude =.git,__pycache__,docs/source/conf.py,build,dist
-ignore = I101,I201,F401,F403,S001,D100,D101,D102,D103,D104,D105,D106,D107,D200,D205,D400,W504,D202,E203,W503,B006
+ignore = I101,I201,F401,F403,S001,D100,D101,D102,D103,D104,D105,D106,D107,D200,D205,D400,W504,D202,E203,E722,W503,B006
 inline-quotes = "
 
 [mypy]