setup.py with installation and console entry point

main.py

@@ -13,15 +13,9 @@
 Usage:
 refer to README.md file
 """
-from sys import exit
-from textwrap import dedent
-from modules.parser import *
-from modules.utils import *
-from modules.downloader import *
-from modules.show import *
-from modules.csv_downloader import *
-from modules.bounding_boxes import *
-from modules.image_level import *
+from oidv4_toolkit.parser import *
+from oidv4_toolkit.bounding_boxes import *
+from oidv4_toolkit.image_level import *
 
 
 ROOT_DIR = ''

oidv4_toolkit/__main__.py

@@ -0,0 +1,23 @@
+import os
+
+from oidv4_toolkit.parser import parser_arguments
+from oidv4_toolkit.bounding_boxes import bounding_boxes_images
+from oidv4_toolkit.image_level import image_level
+
+ROOT_DIR = ''
+DEFAULT_OID_DIR = os.path.join(ROOT_DIR, 'OID')
+
+
+def main():
+
+    args = parser_arguments()
+
+    if args.command == 'downloader_ill':
+        image_level(args, DEFAULT_OID_DIR)
+    else:
+        bounding_boxes_images(args, DEFAULT_OID_DIR)
+
+
+if __name__ == "__main__":
+
+    main()

modules/bounding_boxes.py → oidv4_toolkit/bounding_boxes.py

@@ -1,10 +1,10 @@
 import os
-from modules.utils import *
-from modules.downloader import *
-from modules.show import *
-from modules.csv_downloader import *
+from oidv4_toolkit.utils import *
+from oidv4_toolkit.downloader import *
+from oidv4_toolkit.show import *
+from oidv4_toolkit.csv_downloader import *
 
-from modules.utils import bcolors as bc
+from oidv4_toolkit.utils import bcolors as bc
 
 def bounding_boxes_images(args, DEFAULT_OID_DIR):
 

modules/csv_downloader.py → oidv4_toolkit/csv_downloader.py

@@ -4,7 +4,7 @@
 import urllib.request
 import pandas as pd
 
-from modules.utils import bcolors as bc
+from oidv4_toolkit.utils import bcolors as bc
 
 OID_URL = 'https://storage.googleapis.com/openimages/2018_04/'
 

modules/downloader.py → oidv4_toolkit/downloader.py

@@ -1,10 +1,11 @@
 import os
 import cv2
 from tqdm import tqdm
-from modules.utils import images_options
-from modules.utils import bcolors as bc
+from oidv4_toolkit.utils import images_options
+from oidv4_toolkit.utils import bcolors as bc
 from multiprocessing.dummy import Pool as ThreadPool
 
+
 def download(args, df_val, folder, dataset_dir, class_name, class_code, class_list=None, threads = 20):
     '''
     Manage the download of the images and the label maker.
@@ -18,19 +19,19 @@ def download(args, df_val, folder, dataset_dir, class_name, class_code, class_li
     :param threads: number of threads
     :return: None
     '''
-    if os.name == 'posix':
-        rows, columns = os.popen('stty size', 'r').read().split()
-    elif os.name == 'nt':
-        try:
-            columns, rows = os.get_terminal_size(0)
-        except OSError:
-            columns, rows = os.get_terminal_size(1)
-    else:
-        columns = 50
-    l = int((int(columns) - len(class_name))/2)
-
-    print ('\n' + bc.HEADER + '-'*l + class_name + '-'*l + bc.ENDC)
-    print(bc.INFO + 'Downloading {} images.'.format(args.type_csv) + bc.ENDC)
+    # if os.name == 'posix':
+    #     rows, columns = os.popen('stty size', 'r').read().split()
+    # elif os.name == 'nt':
+    #     try:
+    #         columns, rows = os.get_terminal_size(0)
+    #     except OSError:
+    #         columns, rows = os.get_terminal_size(1)
+    # else:
+    #     columns = 50
+    # l = int((int(columns) - len(class_name))/2)
+    #
+    # print ('\n' + bc.HEADER + '-'*l + class_name + '-'*l + bc.ENDC)
+    # print(bc.INFO + 'Downloading {} images.'.format(args.type_csv) + bc.ENDC)
     df_val_images = images_options(df_val, args)
 
     images_list = df_val_images['ImageID'][df_val_images.LabelName == class_code].values
@@ -74,7 +75,7 @@ def download_img(folder, dataset_dir, class_name, images_list, threads):
         commands = []
         for image in images_list:
             path = image_dir + '/' + str(image) + '.jpg ' + '"' + download_dir + '"'
-            command = 'aws s3 --no-sign-request --only-show-errors cp s3://open-images-dataset/' + path                    
+            command = 'aws s3 --no-sign-request --only-show-errors cp s3://open-images-dataset/' + path
             commands.append(command)
 
         list(tqdm(pool.imap(os.system, commands), total = len(commands) ))