dtlpy 1.115.44__py3-none-any.whl → 1.116.6__py3-none-any.whl

dtlpy/utilities/videos/video_player.py

@@ -1,598 +1,598 @@
-import logging
-import json
-import time
-import os
-import numpy as np
-import dtlpy as dl
-
-logger = logging.getLogger(name='dtlpy')
-
-class VideoPlayer:
-    """
-    Video Player GUI.
-    """
-
-    def __init__(self, project_name=None, project_id=None,
-                 dataset_name=None, dataset_id=None,
-                 item_filepath=None, item_id=None):
-        try:
-            import tkinter
-        except ImportError:
-            logger.error(
-                'Import Error! Cant import tkinter. Annotations operations will be limited. import manually and fix errors')
-            raise
-
-        # init tkinter window
-        self.window = tkinter.Tk()
-        self.window.title('Dataloop Player')
-
-        self.video_source = None
-        self.video_annotations = None
-        self.local_annotations_filename = None
-        self.labels = None
-        self.project = None
-        self.dataset = None
-        self.item = None
-        #############################
-        # load video and annotation #
-        #############################
-        self.platform_params = {'project_name': project_name, 'project_id': project_id,
-                                'dataset_name': dataset_name, 'dataset_id': dataset_id,
-                                'item_filepath': item_filepath, 'item_id': item_id}
-        self.from_platform()
-
-        # load video
-        self.photo = None
-        self.vid = None
-        self.load_video()
-
-        # canvas to put frames on
-        self.canvas = tkinter.Canvas(self.window, width=self.vid.width, height=self.vid.height)
-        self.canvas.pack()
-        self.canvas.bind('<Configure>', self._resize_image)
-
-        ########################
-        # create buttons panel #
-        ########################
-        self.window_width = self.vid.width
-        self.window_height = self.vid.height
-        self.buttons_window = tkinter.Toplevel(master=self.window)
-        self.buttons_window.title('Controls')
-        self.buttons_window.geometry()
-
-        button_frame = tkinter.Frame(self.buttons_window)
-        button_frame.grid(sticky="W", row=0, column=0)
-
-        ###############
-        # information #
-        ###############
-        txt_dataset_name = tkinter.Label(button_frame)
-        txt_dataset_name.grid(sticky="W", row=2, column=0, columnspan=10)
-        txt_dataset_name.configure(text='Dataset name: %s' % self.dataset.name)
-        txt_item_name = tkinter.Label(button_frame)
-        txt_item_name.grid(sticky="W", row=3, column=0, columnspan=10)
-        txt_item_name.configure(text='Item name: %s' % self.item.name)
-        ###############
-        # Exit Button #
-        ###############
-        btn_exit = tkinter.Button(button_frame, text="Close", command=self.close).grid(sticky="W", row=17, column=0)
-        ################
-        # Play / Pause #
-        ################
-        self.btn_toggle_play = tkinter.Button(button_frame, text='Play ', command=self.toggle_play)
-        self.btn_toggle_play.grid(sticky="W", row=4, column=0)
-        ###################
-        # Next prev frame #
-        ###################
-        btn_next_frame = tkinter.Button(button_frame, text="Next frame", command=self.next_frame).grid(sticky="W",
-                                                                                                       row=4,
-                                                                                                       column=2)
-        btn_prev_frame = tkinter.Button(button_frame, text="Prev frame", command=self.prev_frame).grid(sticky="W",
-                                                                                                       row=4,
-                                                                                                       column=3)
-        self.btn_toggle_frame_num = tkinter.Button(button_frame,
-                                                   text="Hide frame number",
-                                                   command=self.toggle_show_frame_number)
-        self.btn_toggle_frame_num.grid(sticky="W", row=5, column=0, columnspan=10)
-        self.btn_toggle_annotations = tkinter.Button(button_frame,
-                                                     text="Hide annotations",
-                                                     command=self.toggle_show_annotations)
-        self.btn_toggle_annotations.grid(sticky="W", row=6, column=0, columnspan=10)
-        self.btn_toggle_label = tkinter.Button(button_frame,
-                                               text="Hide label",
-                                               command=self.toggle_show_label)
-        self.btn_toggle_label.grid(sticky="W", row=7, column=0, columnspan=10)
-        #################
-        # Export button #
-        #################
-        tkinter.Button(button_frame, text="Export video", command=self.export) \
-            .grid(sticky="W", row=16, column=2, columnspan=2)
-        # ################
-        # # Apply button #
-        # ################
-        # btn_apply_offset_fix = tkinter.Button(button_frame, text="Apply offset fix", command=self.apply_offset_fix) \
-        #     .grid(sticky="W", row=16, column=0, columnspan=2)
-        btn_reset = tkinter.Button(button_frame, text="Reset", command=self.reset).grid(sticky="W", row=4, column=4)
-        # #####################
-        # # Annotation Offset #
-        # #####################
-        # tkinter.Label(button_frame, text="Annotation offset (in frames):") \
-        #     .grid(sticky="W", row=13, column=0,
-        #           columnspan=10)
-        # self.annotations_offset_entry = tkinter.Entry(button_frame)
-        # self.annotations_offset_entry.bind("<Return>", self.set_annotation_offset)
-        # self.annotations_offset_entry.grid(sticky="W", row=14, column=0, columnspan=10)
-        # self.annotation_offset_text = tkinter.Label(button_frame)
-        # self.annotation_offset_text.grid(sticky="W", row=15, column=0, columnspan=10)
-        ##############
-        # Set Frames #
-        ##############
-        tkinter.Label(button_frame, text="Set frame (in frames):") \
-            .grid(sticky="W", row=10, column=0, columnspan=10)
-        self.current_frame_entry = tkinter.Entry(button_frame)
-        self.current_frame_entry.bind("<Return>", self.set_current_frame)
-        self.current_frame_entry.grid(sticky="W", row=11, column=0, columnspan=10)
-        self.current_frame_text = tkinter.Label(button_frame)
-        self.current_frame_text.grid(sticky="W", row=12, column=0, columnspan=10)
-        #########################
-        # Timestamp information #
-        #########################
-        self.frame_timestamp_text = tkinter.Label(button_frame)
-        self.frame_timestamp_text.grid(sticky="W", row=8, column=0, columnspan=10)
-        self.annotations_timestamp_text = tkinter.Label(button_frame)
-        self.annotations_timestamp_text.grid(sticky="W", row=9, column=0, columnspan=10)
-
-        ##############
-        # Parameters #
-        ##############
-        self.delay = None
-        self.annotations_timestamp = None
-        # self.annotations_offset = None
-        self.annotations_timestamp = None
-        self.show_frame_num = True
-        self.show_annotations = True
-        self.show_label = True
-        self.playing = False
-        self.init_all_params()
-        self.show_frame(0)
-
-        ###############
-        # Start video #
-        ###############
-        self.delay = int(1000 * 1 / self.vid.fps)
-        self.update()
-        button_frame.lift()
-        self.window.mainloop()
-
-    def _resize_image(self, event):
-        """
-        Resize frame to match window size
-        :param event:
-        :return:
-        """
-        self.window_width = event.width
-        self.window_height = event.height
-
-    def reset(self):
-        """
-        Reset video and annotation
-        :return:
-        """
-        self.from_platform()
-        self.load_video()
-        self.init_all_params()
-        self.show_frame(0)
-
-    def export(self):
-        """
-        Create an annotated video saved to file
-        :return:
-        """
-
-        try:
-            import cv2
-        except (ImportError, ModuleNotFoundError):
-            logger.error(
-                'Import Error! Cant import cv2. Annotations operations will be limited. import manually and fix errors')
-            raise
-
-        try:
-            from tkinter import ttk
-        except ImportError:
-            logger.error(
-                'Import Error! Cant import ttk from tkinter. Annotations operations will be limited. import manually and fix errors')
-            raise
-
-        # start progress bar
-        p, ext = os.path.splitext(self.video_source)
-        output_filename = p + '_out.mp4'
-
-        # read input video
-        reader = cv2.VideoCapture(self.video_source)
-        width = int(reader.get(cv2.CAP_PROP_FRAME_WIDTH))
-        height = int(reader.get(cv2.CAP_PROP_FRAME_HEIGHT))
-        fps = reader.get(cv2.CAP_PROP_FPS)
-        encoding = int(reader.get(cv2.CAP_PROP_FOURCC))
-        n_frames = int(reader.get(cv2.CAP_PROP_FRAME_COUNT))
-        writer = cv2.VideoWriter(output_filename, cv2.VideoWriter_fourcc(*'MP4V'), fps, (width, height))
-
-        # create popup
-        popup = tkinter.Toplevel()
-        tkinter.Label(popup, text='Exporting to\n%s' % output_filename).grid(row=0, column=0)
-        progress_var = tkinter.DoubleVar()
-        progress_bar = ttk.Progressbar(popup, variable=progress_var, maximum=n_frames)
-        progress_bar.grid(row=1, column=0)  # .pack(fill=tk.X, expand=1, side=tk.BOTTOM)
-        popup.pack_slaves()
-
-        i_frame = 0
-        while reader.isOpened():
-            popup.update()
-            ret, frame = reader.read()
-            if not ret:
-                break
-            # mark on frame
-            annotations = [frame_annotation.get_annotation_by_frame(frame=i_frame)
-                           for frame_annotation in self.video_annotations]
-
-            for annotation in annotations:
-                if annotation is None:
-                    continue
-                frame = annotation.show(image=frame,
-                                        color=self.get_class_color(annotation.label))
-                if self.show_label:
-                    text = '%s-%s' % (annotation.label, ','.join(annotation.attributes))
-                    frame = cv2.putText(frame,
-                                        text=text,
-                                        org=tuple([int(np.round(annotation.left)), int(np.round(annotation.top))]),
-                                        color=(255, 0, 0),
-                                        fontFace=cv2.FONT_HERSHEY_DUPLEX,
-                                        fontScale=1,
-                                        thickness=2)
-            # write
-            writer.write(frame)
-            i_frame += 1
-            progress_var.set(i_frame)
-        reader.release()
-        writer.release()
-        popup.destroy()
-
-    def load_video(self):
-        """
-        Load VideoCapture instance
-        :return:
-        """
-        self.vid = VideoCapture(self.video_source)
-
-    def init_all_params(self):
-        """
-        Reset all initial variables
-        :return:
-        """
-        self.annotations_timestamp = 0
-        # self.annotations_offset = 0
-        # self.annotation_offset_text.configure(text='Current: %d' % self.annotations_offset)
-        self.annotations_timestamp_text.configure(text='Annotation timestamp:\n %d' % self.annotations_timestamp)
-        self.annotations_timestamp_text.grid(sticky="W", row=9, column=0, columnspan=10)
-        # set text frames
-        # self.annotations_offset_entry.delete(0, 'end')
-        # self.annotations_offset_entry.insert(0, str(self.annotations_offset))
-        self.current_frame_entry.delete(0, 'end')
-        self.current_frame_entry.insert(0, str(self.vid.frame_number))
-
-    def from_local(self):
-        """
-        Load local video and annotation
-        :return:
-        """
-        if self.local_annotations_filename is not None:
-            with open(self.local_annotations_filename, 'r') as f:
-                data = json.load(f)
-        self.video_annotations = dl.AnnotationCollection.from_json(data['annotations'])
-
-    def from_platform(self):
-        """
-        Load video and annotations from platform
-        :return:
-        """
-        project_name = self.platform_params['project_name']
-        project_id = self.platform_params['project_id']
-        dataset_name = self.platform_params['dataset_name']
-        dataset_id = self.platform_params['dataset_id']
-        item_filepath = self.platform_params['item_filepath']
-        item_id = self.platform_params['item_id']
-
-        # load remote item
-        if dataset_id is None:
-            self.project = dl.projects.get(project_name=project_name, project_id=project_id)
-            if self.project is None:
-                raise ValueError('Project doesnt exists. name: %s, id: %s' % (project_name, project_id))
-            self.dataset = self.project.datasets.get(dataset_name=dataset_name, dataset_id=dataset_id)
-        else:
-            self.dataset = dl.datasets.get(dataset_id=dataset_id)
-        if self.dataset is None:
-            raise ValueError('Dataset doesnt exists. name: %s, id: %s' % (dataset_name, dataset_id))
-        self.item = self.dataset.items.get(filepath=item_filepath, item_id=item_id)
-        if self.item is None:
-            raise ValueError('Item doesnt exists. name: %s, id: %s' % (item_filepath, item_id))
-        self.labels = {label.tag: label.rgb for label in self.dataset.labels}
-        _, ext = os.path.splitext(self.item.filename[1:])
-        video_filename = os.path.join(self.dataset.__get_local_path__(), self.item.filename[1:])
-        if not os.path.isdir(os.path.dirname(video_filename)):
-            os.makedirs(os.path.dirname(video_filename))
-        if not os.path.isfile(video_filename):
-            self.item.download(local_path=os.path.dirname(video_filename), to_items_folder=False)
-        self.video_source = video_filename
-        self.video_annotations = self.item.annotations.list()
-
-    def close(self):
-        """
-        Terminate window and application
-        :return:
-        """
-        self.window.destroy()
-        self.buttons_window.destroy()
-
-    #
-    # def set_annotation_offset(self, entry):
-    #     """
-    #     Set annotations offset to video start time
-    #     :param entry:
-    #     :return:
-    #     """
-    #     self.annotations_offset = int(self.annotations_offset_entry.get())
-    #     self.annotation_offset_text.configure(text='Current: %d' % self.annotations_offset)
-
-    def set_current_frame(self, entry):
-        """
-        Go to specific frame
-        :param entry:
-        :return:
-        """
-        input_frame = int(self.current_frame_entry.get())
-        self.show_frame(frame_number=input_frame)
-
-    def toggle_show_frame_number(self):
-        """
-        Show/hide frame number on frames
-        :return:
-        """
-        if self.show_frame_num:
-            self.show_frame_num = False
-            self.btn_toggle_frame_num.config(text='Show frame num')
-        else:
-            self.show_frame_num = True
-            self.btn_toggle_frame_num.config(text='Hide frame num')
-
-    def toggle_show_annotations(self):
-        """
-        Show/hide annotations from frame
-        :return:
-        """
-        if self.show_annotations:
-            self.show_annotations = False
-            self.btn_toggle_annotations.config(text='Show annotations')
-        else:
-            self.show_annotations = True
-            self.btn_toggle_annotations.config(text='Hide annotations')
-
-    def toggle_show_label(self):
-        """
-        Show/hide label per annotations
-        :return:
-        """
-        if self.show_label:
-            self.show_label = False
-            self.btn_toggle_label.config(text='Show label')
-        else:
-            self.show_label = True
-            self.btn_toggle_label.config(text='Hide label')
-
-    def get_class_color(self, label):
-        """
-        Color of label
-        :param label:
-        :return:
-        """
-        if label not in self.labels:
-            print('[WARNING] label not in dataset labels: %s' % label)
-            return (255, 0, 0)
-        color = self.labels[label]
-        if isinstance(color, str):
-            if color.startswith('rgb'):
-                color = tuple(eval(color.lstrip('rgb')))
-            elif color.startswith('#'):
-                color = tuple(int(color.lstrip('#')[i:i + 2], 16) for i in (0, 2, 4))
-            else:
-                print('[WARNING] Unknown color scheme: %s' % color)
-                color = (255, 0, 0)
-        return color
-
-    def get_annotations(self, frame):
-        """
-        Get all annotations of frame
-        :param frame:
-        :return:
-        """
-        # self.annotations_timestamp = (self.vid.frame_number + self.annotations_offset) / self.vid.fps
-        self.annotations_timestamp = self.vid.frame_number / self.vid.fps
-        frame = self.video_annotations.get_frame(frame_num=self.vid.frame_number).show(image=frame,
-                                                                                       height=frame.shape[0],
-                                                                                       width=frame.shape[1],
-                                                                                       with_text=self.show_label)
-        return frame
-
-    def toggle_play(self):
-        """
-        Toggle play pause
-        :return:
-        """
-        if self.playing:
-            # need to pause
-            self.playing = False
-            self.btn_toggle_play.config(text='Play ')
-        else:
-            # need to play
-            self.playing = True
-            self.btn_toggle_play.config(text='Pause')
-
-    def next_frame(self):
-        """
-        Get next frame
-        :return:
-        """
-        self.show_frame()
-
-    def prev_frame(self):
-        """
-        Get previous frame
-        :return:
-        """
-        self.show_frame(self.vid.frame_number - 1)
-
-    # def apply_offset_fix(self):
-    #     """
-    #     Apply offset to platform
-    #     :return:
-    #     """
-    #     # delete annotations from platform
-    #     for annotation in self.video_annotations.annotations:
-    #         self.item.annotations.delete(annotations_id=annotation.id)
-    #
-    #     # apply annotations fix
-    #     new_annotations = list()
-    #     for annotation in self.video_annotations.annotations:
-    #         new_annotation = annotation.copy()
-    #         for i in range(len(new_annotation['metadata']['snapshots_'])):
-    #             new_annotation['metadata']['snapshots_'][i]['startTime'] -= (self.annotations_offset / self.vid.fps)
-    #             new_annotation['metadata']['snapshots_'][i]['frameNum'] = int(
-    #                 new_annotation['metadata']['snapshots_'][i]['startTime'] * self.vid.fps)
-    #         new_annotations.append(json.dumps(new_annotation))
-    #
-    #     # upload annotations
-    #     self.item.annotation.upload(annotation=[new_annotations])
-
-    def show_frame(self, frame_number=None):
-        """
-        Get a frame from the video source
-        :param frame_number:
-        :return:
-        """
-        try:
-            import cv2
-        except (ImportError, ModuleNotFoundError):
-            logger.error(
-                'Import Error! Cant import cv2. Annotations operations will be limited. import manually and fix errors')
-            raise
-
-        try:
-            import tkinter
-        except ImportError:
-            logger.error(
-                'Import Error! Cant import tkinter. Annotations operations will be limited. import manually and fix errors')
-            raise
-
-        try:
-            import PIL.ImageTk
-            import PIL.Image
-        except ImportError:
-            logger.error(
-                'Import Error! Cant import PIL.ImageTk/PIL.ImageTk. Annotations operations will be limited. '
-                'import manually and fix errors')
-            raise
-        ret, frame = self.vid.get_frame(frame_number)
-        if ret:
-            if self.show_annotations:
-                frame = self.get_annotations(frame)
-            if self.show_frame_num:
-                text = '%d - %d' % (self.vid.frame_number, np.round(self.annotations_timestamp * self.vid.fps))
-                frame = cv2.putText(frame,
-                                    text=text,
-                                    org=(100, 100),
-                                    color=(0, 0, 255),
-                                    fontFace=cv2.FONT_HERSHEY_DUPLEX,
-                                    fontScale=2,
-                                    thickness=3)
-
-            self.photo = PIL.ImageTk.PhotoImage(
-                image=PIL.Image.fromarray(frame).resize((self.window_width, self.window_height)),
-            master=self.canvas)
-            self.canvas.create_image(0, 0, image=self.photo, anchor=tkinter.NW)
-            # set timestamp
-            self.current_frame_text.configure(text='Frame number:\n%d' % self.vid.frame_number)
-            self.current_frame_text.grid(sticky="W", row=12, column=0, columnspan=10)
-            millis = int(1000 * self.vid.frame_number / self.vid.fps)
-            seconds = (millis / 1000) % 60
-            minutes = int((millis / (1000 * 60)) % 60)
-            hours = int((millis / (1000 * 60 * 60)) % 24)
-            self.frame_timestamp_text.configure(
-                text='Frame timestamp:\n{:02d}:{:02d}:{:.3f}'.format(hours, minutes, seconds))
-            self.frame_timestamp_text.grid(sticky="W", row=8, column=0, columnspan=10)
-
-    def update(self):
-        """
-        Playing the movie. Get frames at the FPS and show
-        :return:
-        """
-        tic = time.time()
-        if self.playing:
-            self.show_frame()
-        add_delay = np.maximum(1, self.delay - int(1000 * (time.time() - tic)))
-        self.window.after(ms=add_delay, func=self.update)
-
-
-class VideoCapture:
-    """
-    Video class using cv2 to play video
-    """
-    def __init__(self, source=0):
-        try:
-            import cv2
-        except (ImportError, ModuleNotFoundError):
-            logger.error(
-                'Import Error! Cant import cv2. Annotations operations will be limited. import manually and fix errors')
-            raise
-        # Open the video source
-        self.vid = cv2.VideoCapture(source)
-        if not self.vid.isOpened():
-            raise ValueError("Unable to open video source", source)
-
-        # Get video source width and height
-        self.width = self.vid.get(cv2.CAP_PROP_FRAME_WIDTH)
-        self.height = self.vid.get(cv2.CAP_PROP_FRAME_HEIGHT)
-        self.fps = self.vid.get(cv2.CAP_PROP_FPS)
-        self.frame_number = self.vid.get(cv2.CAP_PROP_POS_FRAMES)
-
-    def get_frame(self, frame_number=None):
-        """
-        get the frame from video
-        :param frame_number:
-        :return:
-        """
-        try:
-            import cv2
-        except (ImportError, ModuleNotFoundError):
-            logger.error(
-                'Import Error! Cant import cv2. Annotations operations will be limited. import manually and fix errors')
-            raise
-        if self.vid.isOpened():
-            if self.frame_number is None:
-                self.frame_number = self.vid.get(cv2.CAP_PROP_POS_FRAMES)
-            else:
-                self.frame_number += 1
-            if frame_number is not None:
-                self.frame_number = frame_number
-                self.vid.set(cv2.CAP_PROP_POS_FRAMES, frame_number)
-            ret, frame = self.vid.read()
-
-            if ret:
-                # Return a boolean success flag and the current frame converted to BGR
-                return ret, cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-            else:
-                return ret, None
-        else:
-            return False, None
-
-    # Release the video source when the object is destroyed
-    def __del__(self):
-        if self.vid.isOpened():
-            self.vid.release()
+import logging
+import json
+import time
+import os
+import numpy as np
+import dtlpy as dl
+
+logger = logging.getLogger(name='dtlpy')
+
+class VideoPlayer:
+    """
+    Video Player GUI.
+    """
+
+    def __init__(self, project_name=None, project_id=None,
+                 dataset_name=None, dataset_id=None,
+                 item_filepath=None, item_id=None):
+        try:
+            import tkinter
+        except ImportError:
+            logger.error(
+                'Import Error! Cant import tkinter. Annotations operations will be limited. import manually and fix errors')
+            raise
+
+        # init tkinter window
+        self.window = tkinter.Tk()
+        self.window.title('Dataloop Player')
+
+        self.video_source = None
+        self.video_annotations = None
+        self.local_annotations_filename = None
+        self.labels = None
+        self.project = None
+        self.dataset = None
+        self.item = None
+        #############################
+        # load video and annotation #
+        #############################
+        self.platform_params = {'project_name': project_name, 'project_id': project_id,
+                                'dataset_name': dataset_name, 'dataset_id': dataset_id,
+                                'item_filepath': item_filepath, 'item_id': item_id}
+        self.from_platform()
+
+        # load video
+        self.photo = None
+        self.vid = None
+        self.load_video()
+
+        # canvas to put frames on
+        self.canvas = tkinter.Canvas(self.window, width=self.vid.width, height=self.vid.height)
+        self.canvas.pack()
+        self.canvas.bind('<Configure>', self._resize_image)
+
+        ########################
+        # create buttons panel #
+        ########################
+        self.window_width = self.vid.width
+        self.window_height = self.vid.height
+        self.buttons_window = tkinter.Toplevel(master=self.window)
+        self.buttons_window.title('Controls')
+        self.buttons_window.geometry()
+
+        button_frame = tkinter.Frame(self.buttons_window)
+        button_frame.grid(sticky="W", row=0, column=0)
+
+        ###############
+        # information #
+        ###############
+        txt_dataset_name = tkinter.Label(button_frame)
+        txt_dataset_name.grid(sticky="W", row=2, column=0, columnspan=10)
+        txt_dataset_name.configure(text='Dataset name: %s' % self.dataset.name)
+        txt_item_name = tkinter.Label(button_frame)
+        txt_item_name.grid(sticky="W", row=3, column=0, columnspan=10)
+        txt_item_name.configure(text='Item name: %s' % self.item.name)
+        ###############
+        # Exit Button #
+        ###############
+        btn_exit = tkinter.Button(button_frame, text="Close", command=self.close).grid(sticky="W", row=17, column=0)
+        ################
+        # Play / Pause #
+        ################
+        self.btn_toggle_play = tkinter.Button(button_frame, text='Play ', command=self.toggle_play)
+        self.btn_toggle_play.grid(sticky="W", row=4, column=0)
+        ###################
+        # Next prev frame #
+        ###################
+        btn_next_frame = tkinter.Button(button_frame, text="Next frame", command=self.next_frame).grid(sticky="W",
+                                                                                                       row=4,
+                                                                                                       column=2)
+        btn_prev_frame = tkinter.Button(button_frame, text="Prev frame", command=self.prev_frame).grid(sticky="W",
+                                                                                                       row=4,
+                                                                                                       column=3)
+        self.btn_toggle_frame_num = tkinter.Button(button_frame,
+                                                   text="Hide frame number",
+                                                   command=self.toggle_show_frame_number)
+        self.btn_toggle_frame_num.grid(sticky="W", row=5, column=0, columnspan=10)
+        self.btn_toggle_annotations = tkinter.Button(button_frame,
+                                                     text="Hide annotations",
+                                                     command=self.toggle_show_annotations)
+        self.btn_toggle_annotations.grid(sticky="W", row=6, column=0, columnspan=10)
+        self.btn_toggle_label = tkinter.Button(button_frame,
+                                               text="Hide label",
+                                               command=self.toggle_show_label)
+        self.btn_toggle_label.grid(sticky="W", row=7, column=0, columnspan=10)
+        #################
+        # Export button #
+        #################
+        tkinter.Button(button_frame, text="Export video", command=self.export) \
+            .grid(sticky="W", row=16, column=2, columnspan=2)
+        # ################
+        # # Apply button #
+        # ################
+        # btn_apply_offset_fix = tkinter.Button(button_frame, text="Apply offset fix", command=self.apply_offset_fix) \
+        #     .grid(sticky="W", row=16, column=0, columnspan=2)
+        btn_reset = tkinter.Button(button_frame, text="Reset", command=self.reset).grid(sticky="W", row=4, column=4)
+        # #####################
+        # # Annotation Offset #
+        # #####################
+        # tkinter.Label(button_frame, text="Annotation offset (in frames):") \
+        #     .grid(sticky="W", row=13, column=0,
+        #           columnspan=10)
+        # self.annotations_offset_entry = tkinter.Entry(button_frame)
+        # self.annotations_offset_entry.bind("<Return>", self.set_annotation_offset)
+        # self.annotations_offset_entry.grid(sticky="W", row=14, column=0, columnspan=10)
+        # self.annotation_offset_text = tkinter.Label(button_frame)
+        # self.annotation_offset_text.grid(sticky="W", row=15, column=0, columnspan=10)
+        ##############
+        # Set Frames #
+        ##############
+        tkinter.Label(button_frame, text="Set frame (in frames):") \
+            .grid(sticky="W", row=10, column=0, columnspan=10)
+        self.current_frame_entry = tkinter.Entry(button_frame)
+        self.current_frame_entry.bind("<Return>", self.set_current_frame)
+        self.current_frame_entry.grid(sticky="W", row=11, column=0, columnspan=10)
+        self.current_frame_text = tkinter.Label(button_frame)
+        self.current_frame_text.grid(sticky="W", row=12, column=0, columnspan=10)
+        #########################
+        # Timestamp information #
+        #########################
+        self.frame_timestamp_text = tkinter.Label(button_frame)
+        self.frame_timestamp_text.grid(sticky="W", row=8, column=0, columnspan=10)
+        self.annotations_timestamp_text = tkinter.Label(button_frame)
+        self.annotations_timestamp_text.grid(sticky="W", row=9, column=0, columnspan=10)
+
+        ##############
+        # Parameters #
+        ##############
+        self.delay = None
+        self.annotations_timestamp = None
+        # self.annotations_offset = None
+        self.annotations_timestamp = None
+        self.show_frame_num = True
+        self.show_annotations = True
+        self.show_label = True
+        self.playing = False
+        self.init_all_params()
+        self.show_frame(0)
+
+        ###############
+        # Start video #
+        ###############
+        self.delay = int(1000 * 1 / self.vid.fps)
+        self.update()
+        button_frame.lift()
+        self.window.mainloop()
+
+    def _resize_image(self, event):
+        """
+        Resize frame to match window size
+        :param event:
+        :return:
+        """
+        self.window_width = event.width
+        self.window_height = event.height
+
+    def reset(self):
+        """
+        Reset video and annotation
+        :return:
+        """
+        self.from_platform()
+        self.load_video()
+        self.init_all_params()
+        self.show_frame(0)
+
+    def export(self):
+        """
+        Create an annotated video saved to file
+        :return:
+        """
+
+        try:
+            import cv2
+        except (ImportError, ModuleNotFoundError):
+            logger.error(
+                'Import Error! Cant import cv2. Annotations operations will be limited. import manually and fix errors')
+            raise
+
+        try:
+            from tkinter import ttk
+        except ImportError:
+            logger.error(
+                'Import Error! Cant import ttk from tkinter. Annotations operations will be limited. import manually and fix errors')
+            raise
+
+        # start progress bar
+        p, ext = os.path.splitext(self.video_source)
+        output_filename = p + '_out.mp4'
+
+        # read input video
+        reader = cv2.VideoCapture(self.video_source)
+        width = int(reader.get(cv2.CAP_PROP_FRAME_WIDTH))
+        height = int(reader.get(cv2.CAP_PROP_FRAME_HEIGHT))
+        fps = reader.get(cv2.CAP_PROP_FPS)
+        encoding = int(reader.get(cv2.CAP_PROP_FOURCC))
+        n_frames = int(reader.get(cv2.CAP_PROP_FRAME_COUNT))
+        writer = cv2.VideoWriter(output_filename, cv2.VideoWriter_fourcc(*'MP4V'), fps, (width, height))
+
+        # create popup
+        popup = tkinter.Toplevel()
+        tkinter.Label(popup, text='Exporting to\n%s' % output_filename).grid(row=0, column=0)
+        progress_var = tkinter.DoubleVar()
+        progress_bar = ttk.Progressbar(popup, variable=progress_var, maximum=n_frames)
+        progress_bar.grid(row=1, column=0)  # .pack(fill=tk.X, expand=1, side=tk.BOTTOM)
+        popup.pack_slaves()
+
+        i_frame = 0
+        while reader.isOpened():
+            popup.update()
+            ret, frame = reader.read()
+            if not ret:
+                break
+            # mark on frame
+            annotations = [frame_annotation.get_annotation_by_frame(frame=i_frame)
+                           for frame_annotation in self.video_annotations]
+
+            for annotation in annotations:
+                if annotation is None:
+                    continue
+                frame = annotation.show(image=frame,
+                                        color=self.get_class_color(annotation.label))
+                if self.show_label:
+                    text = '%s-%s' % (annotation.label, ','.join(annotation.attributes))
+                    frame = cv2.putText(frame,
+                                        text=text,
+                                        org=tuple([int(np.round(annotation.left)), int(np.round(annotation.top))]),
+                                        color=(255, 0, 0),
+                                        fontFace=cv2.FONT_HERSHEY_DUPLEX,
+                                        fontScale=1,
+                                        thickness=2)
+            # write
+            writer.write(frame)
+            i_frame += 1
+            progress_var.set(i_frame)
+        reader.release()
+        writer.release()
+        popup.destroy()
+
+    def load_video(self):
+        """
+        Load VideoCapture instance
+        :return:
+        """
+        self.vid = VideoCapture(self.video_source)
+
+    def init_all_params(self):
+        """
+        Reset all initial variables
+        :return:
+        """
+        self.annotations_timestamp = 0
+        # self.annotations_offset = 0
+        # self.annotation_offset_text.configure(text='Current: %d' % self.annotations_offset)
+        self.annotations_timestamp_text.configure(text='Annotation timestamp:\n %d' % self.annotations_timestamp)
+        self.annotations_timestamp_text.grid(sticky="W", row=9, column=0, columnspan=10)
+        # set text frames
+        # self.annotations_offset_entry.delete(0, 'end')
+        # self.annotations_offset_entry.insert(0, str(self.annotations_offset))
+        self.current_frame_entry.delete(0, 'end')
+        self.current_frame_entry.insert(0, str(self.vid.frame_number))
+
+    def from_local(self):
+        """
+        Load local video and annotation
+        :return:
+        """
+        if self.local_annotations_filename is not None:
+            with open(self.local_annotations_filename, 'r') as f:
+                data = json.load(f)
+        self.video_annotations = dl.AnnotationCollection.from_json(data['annotations'])
+
+    def from_platform(self):
+        """
+        Load video and annotations from platform
+        :return:
+        """
+        project_name = self.platform_params['project_name']
+        project_id = self.platform_params['project_id']
+        dataset_name = self.platform_params['dataset_name']
+        dataset_id = self.platform_params['dataset_id']
+        item_filepath = self.platform_params['item_filepath']
+        item_id = self.platform_params['item_id']
+
+        # load remote item
+        if dataset_id is None:
+            self.project = dl.projects.get(project_name=project_name, project_id=project_id)
+            if self.project is None:
+                raise ValueError('Project doesnt exists. name: %s, id: %s' % (project_name, project_id))
+            self.dataset = self.project.datasets.get(dataset_name=dataset_name, dataset_id=dataset_id)
+        else:
+            self.dataset = dl.datasets.get(dataset_id=dataset_id)
+        if self.dataset is None:
+            raise ValueError('Dataset doesnt exists. name: %s, id: %s' % (dataset_name, dataset_id))
+        self.item = self.dataset.items.get(filepath=item_filepath, item_id=item_id)
+        if self.item is None:
+            raise ValueError('Item doesnt exists. name: %s, id: %s' % (item_filepath, item_id))
+        self.labels = {label.tag: label.rgb for label in self.dataset.labels}
+        _, ext = os.path.splitext(self.item.filename[1:])
+        video_filename = os.path.join(self.dataset.__get_local_path__(), self.item.filename[1:])
+        if not os.path.isdir(os.path.dirname(video_filename)):
+            os.makedirs(os.path.dirname(video_filename))
+        if not os.path.isfile(video_filename):
+            self.item.download(local_path=os.path.dirname(video_filename), to_items_folder=False)
+        self.video_source = video_filename
+        self.video_annotations = self.item.annotations.list()
+
+    def close(self):
+        """
+        Terminate window and application
+        :return:
+        """
+        self.window.destroy()
+        self.buttons_window.destroy()
+
+    #
+    # def set_annotation_offset(self, entry):
+    #     """
+    #     Set annotations offset to video start time
+    #     :param entry:
+    #     :return:
+    #     """
+    #     self.annotations_offset = int(self.annotations_offset_entry.get())
+    #     self.annotation_offset_text.configure(text='Current: %d' % self.annotations_offset)
+
+    def set_current_frame(self, entry):
+        """
+        Go to specific frame
+        :param entry:
+        :return:
+        """
+        input_frame = int(self.current_frame_entry.get())
+        self.show_frame(frame_number=input_frame)
+
+    def toggle_show_frame_number(self):
+        """
+        Show/hide frame number on frames
+        :return:
+        """
+        if self.show_frame_num:
+            self.show_frame_num = False
+            self.btn_toggle_frame_num.config(text='Show frame num')
+        else:
+            self.show_frame_num = True
+            self.btn_toggle_frame_num.config(text='Hide frame num')
+
+    def toggle_show_annotations(self):
+        """
+        Show/hide annotations from frame
+        :return:
+        """
+        if self.show_annotations:
+            self.show_annotations = False
+            self.btn_toggle_annotations.config(text='Show annotations')
+        else:
+            self.show_annotations = True
+            self.btn_toggle_annotations.config(text='Hide annotations')
+
+    def toggle_show_label(self):
+        """
+        Show/hide label per annotations
+        :return:
+        """
+        if self.show_label:
+            self.show_label = False
+            self.btn_toggle_label.config(text='Show label')
+        else:
+            self.show_label = True
+            self.btn_toggle_label.config(text='Hide label')
+
+    def get_class_color(self, label):
+        """
+        Color of label
+        :param label:
+        :return:
+        """
+        if label not in self.labels:
+            print('[WARNING] label not in dataset labels: %s' % label)
+            return (255, 0, 0)
+        color = self.labels[label]
+        if isinstance(color, str):
+            if color.startswith('rgb'):
+                color = tuple(eval(color.lstrip('rgb')))
+            elif color.startswith('#'):
+                color = tuple(int(color.lstrip('#')[i:i + 2], 16) for i in (0, 2, 4))
+            else:
+                print('[WARNING] Unknown color scheme: %s' % color)
+                color = (255, 0, 0)
+        return color
+
+    def get_annotations(self, frame):
+        """
+        Get all annotations of frame
+        :param frame:
+        :return:
+        """
+        # self.annotations_timestamp = (self.vid.frame_number + self.annotations_offset) / self.vid.fps
+        self.annotations_timestamp = self.vid.frame_number / self.vid.fps
+        frame = self.video_annotations.get_frame(frame_num=self.vid.frame_number).show(image=frame,
+                                                                                       height=frame.shape[0],
+                                                                                       width=frame.shape[1],
+                                                                                       with_text=self.show_label)
+        return frame
+
+    def toggle_play(self):
+        """
+        Toggle play pause
+        :return:
+        """
+        if self.playing:
+            # need to pause
+            self.playing = False
+            self.btn_toggle_play.config(text='Play ')
+        else:
+            # need to play
+            self.playing = True
+            self.btn_toggle_play.config(text='Pause')
+
+    def next_frame(self):
+        """
+        Get next frame
+        :return:
+        """
+        self.show_frame()
+
+    def prev_frame(self):
+        """
+        Get previous frame
+        :return:
+        """
+        self.show_frame(self.vid.frame_number - 1)
+
+    # def apply_offset_fix(self):
+    #     """
+    #     Apply offset to platform
+    #     :return:
+    #     """
+    #     # delete annotations from platform
+    #     for annotation in self.video_annotations.annotations:
+    #         self.item.annotations.delete(annotations_id=annotation.id)
+    #
+    #     # apply annotations fix
+    #     new_annotations = list()
+    #     for annotation in self.video_annotations.annotations:
+    #         new_annotation = annotation.copy()
+    #         for i in range(len(new_annotation['metadata']['snapshots_'])):
+    #             new_annotation['metadata']['snapshots_'][i]['startTime'] -= (self.annotations_offset / self.vid.fps)
+    #             new_annotation['metadata']['snapshots_'][i]['frameNum'] = int(
+    #                 new_annotation['metadata']['snapshots_'][i]['startTime'] * self.vid.fps)
+    #         new_annotations.append(json.dumps(new_annotation))
+    #
+    #     # upload annotations
+    #     self.item.annotation.upload(annotation=[new_annotations])
+
+    def show_frame(self, frame_number=None):
+        """
+        Get a frame from the video source
+        :param frame_number:
+        :return:
+        """
+        try:
+            import cv2
+        except (ImportError, ModuleNotFoundError):
+            logger.error(
+                'Import Error! Cant import cv2. Annotations operations will be limited. import manually and fix errors')
+            raise
+
+        try:
+            import tkinter
+        except ImportError:
+            logger.error(
+                'Import Error! Cant import tkinter. Annotations operations will be limited. import manually and fix errors')
+            raise
+
+        try:
+            import PIL.ImageTk
+            import PIL.Image
+        except ImportError:
+            logger.error(
+                'Import Error! Cant import PIL.ImageTk/PIL.ImageTk. Annotations operations will be limited. '
+                'import manually and fix errors')
+            raise
+        ret, frame = self.vid.get_frame(frame_number)
+        if ret:
+            if self.show_annotations:
+                frame = self.get_annotations(frame)
+            if self.show_frame_num:
+                text = '%d - %d' % (self.vid.frame_number, np.round(self.annotations_timestamp * self.vid.fps))
+                frame = cv2.putText(frame,
+                                    text=text,
+                                    org=(100, 100),
+                                    color=(0, 0, 255),
+                                    fontFace=cv2.FONT_HERSHEY_DUPLEX,
+                                    fontScale=2,
+                                    thickness=3)
+
+            self.photo = PIL.ImageTk.PhotoImage(
+                image=PIL.Image.fromarray(frame).resize((self.window_width, self.window_height)),
+            master=self.canvas)
+            self.canvas.create_image(0, 0, image=self.photo, anchor=tkinter.NW)
+            # set timestamp
+            self.current_frame_text.configure(text='Frame number:\n%d' % self.vid.frame_number)
+            self.current_frame_text.grid(sticky="W", row=12, column=0, columnspan=10)
+            millis = int(1000 * self.vid.frame_number / self.vid.fps)
+            seconds = (millis / 1000) % 60
+            minutes = int((millis / (1000 * 60)) % 60)
+            hours = int((millis / (1000 * 60 * 60)) % 24)
+            self.frame_timestamp_text.configure(
+                text='Frame timestamp:\n{:02d}:{:02d}:{:.3f}'.format(hours, minutes, seconds))
+            self.frame_timestamp_text.grid(sticky="W", row=8, column=0, columnspan=10)
+
+    def update(self):
+        """
+        Playing the movie. Get frames at the FPS and show
+        :return:
+        """
+        tic = time.time()
+        if self.playing:
+            self.show_frame()
+        add_delay = np.maximum(1, self.delay - int(1000 * (time.time() - tic)))
+        self.window.after(ms=add_delay, func=self.update)
+
+
+class VideoCapture:
+    """
+    Video class using cv2 to play video
+    """
+    def __init__(self, source=0):
+        try:
+            import cv2
+        except (ImportError, ModuleNotFoundError):
+            logger.error(
+                'Import Error! Cant import cv2. Annotations operations will be limited. import manually and fix errors')
+            raise
+        # Open the video source
+        self.vid = cv2.VideoCapture(source)
+        if not self.vid.isOpened():
+            raise ValueError("Unable to open video source", source)
+
+        # Get video source width and height
+        self.width = self.vid.get(cv2.CAP_PROP_FRAME_WIDTH)
+        self.height = self.vid.get(cv2.CAP_PROP_FRAME_HEIGHT)
+        self.fps = self.vid.get(cv2.CAP_PROP_FPS)
+        self.frame_number = self.vid.get(cv2.CAP_PROP_POS_FRAMES)
+
+    def get_frame(self, frame_number=None):
+        """
+        get the frame from video
+        :param frame_number:
+        :return:
+        """
+        try:
+            import cv2
+        except (ImportError, ModuleNotFoundError):
+            logger.error(
+                'Import Error! Cant import cv2. Annotations operations will be limited. import manually and fix errors')
+            raise
+        if self.vid.isOpened():
+            if self.frame_number is None:
+                self.frame_number = self.vid.get(cv2.CAP_PROP_POS_FRAMES)
+            else:
+                self.frame_number += 1
+            if frame_number is not None:
+                self.frame_number = frame_number
+                self.vid.set(cv2.CAP_PROP_POS_FRAMES, frame_number)
+            ret, frame = self.vid.read()
+
+            if ret:
+                # Return a boolean success flag and the current frame converted to BGR
+                return ret, cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            else:
+                return ret, None
+        else:
+            return False, None
+
+    # Release the video source when the object is destroyed
+    def __del__(self):
+        if self.vid.isOpened():
+            self.vid.release()