Add upload videos cli

Author: hovnatan

docs/source/cli.rst

@@ -52,6 +52,38 @@ scanned for available images.
 
 Optional argument *extensions* accepts comma separated list of image extensions to look for. If the argument is not given then value *jpg,png* is assumed.
 
+----------
+
+.. _ref_upload_videos:
+
+Uploading videos
+~~~~~~~~~~~~~~~~
+
+To upload videos from folder to project use:
+
+.. code-block:: bash
+
+   superannotate upload-videos --project <project_name> --folder <folder_path> 
+                               [--recursive] [--extensions mp4,avi,mov,webm,flv,mpg,ogg]
+                               [--target-fps <float>] [--start-time <float>]
+                               [--end-time <float>]
+
+If optional argument *recursive* is given then subfolders of :file:`<folder_path>` are also recursively
+scanned for available images.
+
+Optional argument *extensions* accepts comma separated list of image extensions
+to look for. If the argument is not given then value *mp4,avi,mov,webm,flv,mpg,ogg* is assumed.
+
+*target-fps* specifies how many frames per second need to extract from the videos (approximate).
+If not specified all frames will be uploaded.
+
+*start-time* specifies time (in seconds) from which to start extracting frames,
+default is 0.0.
+
+*end-time* specifies time (in seconds) up to which to extract frames. 
+If it is not specified, then up to end is assumed.
+
+
 ----------
 
 .. _ref_cli_version:

superannotate/__main__.py

@@ -43,8 +43,10 @@ def main():
     command = sys.argv[1]
     further_args = sys.argv[2:]
 
-    if command == "image-upload":
+    if command == "upload-images":
         image_upload(further_args)
+    elif command == "upload-videos":
+        video_upload(further_args)
     elif command == "init":
         ask_token(further_args)
     elif command == "export-project":
@@ -59,6 +61,70 @@ def _list_str(values):
     return values.split(',')
 
 
+def video_upload(args):
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        '--project', required=True, help='Project name to upload'
+    )
+    parser.add_argument(
+        '--folder', required=True, help='Folder from which to upload'
+    )
+    parser.add_argument(
+        '--recursive',
+        default=False,
+        action='store_true',
+        help='Enables recursive subfolder upload.'
+    )
+    parser.add_argument(
+        '--extensions',
+        required=False,
+        default=None,
+        type=_list_str,
+        help=
+        'List of video extensions to include. Default is mp4,avi,mov,webm,flv,mpg,ogg'
+    )
+    parser.add_argument(
+        '--set-annotation-status',
+        required=False,
+        default="NotStarted",
+        help=
+        'Set images\' annotation statuses after upload. Default is NotStarted'
+    )
+    parser.add_argument(
+        '--target-fps',
+        required=False,
+        default=None,
+        help=
+        'How many frames per second need to extract from the videos (approximate).  If not specified all frames will be uploaded'
+    )
+    parser.add_argument(
+        '--start-time',
+        required=False,
+        default=0.0,
+        help=
+        'Time (in seconds) from which to start extracting frames. Default is 0.0'
+    )
+    parser.add_argument(
+        '--end-time',
+        required=False,
+        default=None,
+        help=
+        'Time (in seconds) up to which to extract frames. If it is not specified, then up to end'
+    )
+    args = parser.parse_args(args)
+
+    sa.upload_videos_from_folder_to_project(
+        project=args.project,
+        folder_path=args.folder,
+        extensions=args.extensions,
+        annotation_status=args.set_annotation_status,
+        recursive_subfolders=args.recursive,
+        target_fps=args.target_fps,
+        start_time=args.start_time,
+        end_time=args.end_time
+    )
+
+
 def image_upload(args):
     parser = argparse.ArgumentParser()
     parser.add_argument(

superannotate/db/projects.py

@@ -413,8 +413,9 @@ def upload_videos_from_folder_to_project(
     :type end_time: float
     :param annotation_status: value to set the annotation statuses of the uploaded images NotStarted InProgress QualityCheck Returned Completed Skipped
     :type annotation_status: str
-    :param image_quality_in_editor: image quality (in percents) that will be seen in SuperAnnotate web annotation editor. If None default value will be used.
-    :type image_quality_in_editor: int
+    :param image_quality_in_editor: image quality be seen in SuperAnnotate web annotation editor.
+           Can be either "compressed" or "original".  If None then the default value in project settings will be used.
+    :type image_quality_in_editor: str
 
     :return: uploaded and not-uploaded video frame images' filenames
     :rtype: tuple of list of strs