simba-uw-tf-dev 4.7.6__py3-none-any.whl → 4.7.8__py3-none-any.whl

simba/ui/pop_ups/sleap_csv_predictions_to_yolo_popup.py

@@ -21,7 +21,8 @@ TRAIN_SIZE_OPTIONS = np.arange(10, 110, 10)
 SAMPLE_SIZE_OPTIONS = list(np.arange(50, 650, 50))
 
 PADDING_OPTIONS = list(np.round(np.arange(0.01, 10.05, 0.05),2).astype(str))
-PADDING_OPTIONS = np.insert(PADDING_OPTIONS, 0, 'None')
+PADDING_OPTIONS = list(np.insert(PADDING_OPTIONS, 0, 'None'))
+
 
 THRESHOLD_OPTION = list(range(10, 110, 10))
 
@@ -38,19 +39,19 @@ class SLEAPcsvInference2Yolo(PopUpMixin):
     def __init__(self):
         PopUpMixin.__init__(self, title="SLEAP CSV PREDICTIONS TO YOLO POSE ESTIMATION ANNOTATIONS", icon='sleap_small')
         settings_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="SETTINGS", icon_name='settings')
-        self.sleap_dir = FolderSelect(settings_frm, folderDescription="SLEAP DATA DIRECTORY:", lblwidth=35, entry_width=40, initialdir=r"D:\troubleshooting\two_animals_sleap\import_data", lbl_icon='folder')
-        self.video_dir = FolderSelect(settings_frm, folderDescription="VIDEO DIRECTORY:", lblwidth=35, entry_width=40, initialdir=r"D:\troubleshooting\two_animals_sleap\project_folder\videos", lbl_icon='folder')
-        self.save_dir = FolderSelect(settings_frm, folderDescription="SAVE DIRECTORY:", lblwidth=35, entry_width=40, initialdir=r"D:\troubleshooting\two_animals_sleap\yolo_kpts_2", lbl_icon='folder')
-
-        self.frm_cnt_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=SAMPLE_SIZE_OPTIONS, label="FRAMES (PER VIDEO): ", label_width=35, dropdown_width=40, value=100, img='frames')
-        self.verbose_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="VERBOSE: ", label_width=35, dropdown_width=40, value='TRUE', img='frames')
-        self.threshold_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=THRESHOLD_OPTION, label="THRESHOLD (%): ", label_width=35, dropdown_width=40, value=90, img='threshold')
-        self.train_size_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=TRAIN_SIZE_OPTIONS, label="TRAIN SIZE (%): ", label_width=35, dropdown_width=40, value=70, img='pct_2')
-        self.grey_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="GREYSCALE: ", label_width=35, dropdown_width=40, value='FALSE', img='grey')
-        self.padding_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=PADDING_OPTIONS, label="PADDING: ", label_width=35, dropdown_width=40, value='None', img='resize')
-        self.animal_cnt_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=list(range(1, 10, 1)), label="ANIMAL COUNT: ", label_width=35, dropdown_width=40, value=2, img='abacus_2')
-        self.clahe_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="CLAHE: ", label_width=35, dropdown_width=40, value='FALSE', img='clahe')
-        self.single_id_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="REMOVE ANIMAL ID'S", label_width=35, dropdown_width=40, value='FALSE', img='mouse_head')
+        self.sleap_dir = FolderSelect(settings_frm, folderDescription="SLEAP DATA DIRECTORY:", lblwidth=35, entry_width=40, initialdir=r"D:\troubleshooting\two_animals_sleap\import_data", lbl_icon='folder', tooltip_key='SLEAP_DATA_DIR')
+        self.video_dir = FolderSelect(settings_frm, folderDescription="VIDEO DIRECTORY:", lblwidth=35, entry_width=40, initialdir=r"D:\troubleshooting\two_animals_sleap\project_folder\videos", lbl_icon='folder_video', tooltip_key='VIDEO_DIR')
+        self.save_dir = FolderSelect(settings_frm, folderDescription="SAVE DIRECTORY:", lblwidth=35, entry_width=40, initialdir=r"D:\troubleshooting\two_animals_sleap\yolo_kpts_2", lbl_icon='folder_2', tooltip_key='SAVE_DIR')
+
+        self.frm_cnt_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=SAMPLE_SIZE_OPTIONS, label="FRAMES (PER VIDEO): ", label_width=35, dropdown_width=40, value=100, img='frames', tooltip_key='simba2yolo_sample_size')
+        self.verbose_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="VERBOSE: ", label_width=35, dropdown_width=40, value='TRUE', img='verbose', tooltip_key='verbose_dropdown')
+        self.threshold_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=THRESHOLD_OPTION, label="THRESHOLD (%): ", label_width=35, dropdown_width=40, value=90, img='threshold', tooltip_key='sleap_threshold')
+        self.train_size_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=TRAIN_SIZE_OPTIONS, label="TRAIN SIZE (%): ", label_width=35, dropdown_width=40, value=70, img='pct_2', tooltip_key='simba2yolo_train_size')
+        self.grey_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="GREYSCALE: ", label_width=35, dropdown_width=40, value='FALSE', img='grey', tooltip_key='simba2yolo_grey')
+        self.padding_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=PADDING_OPTIONS, label="PADDING: ", label_width=35, dropdown_width=40, value='None', img='resize', tooltip_key='simba2yolo_padding')
+        self.animal_cnt_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=list(range(1, 10, 1)), label="ANIMAL COUNT: ", label_width=35, dropdown_width=40, value=2, img='abacus_2', tooltip_key='ANIMAL_COUNT')
+        self.clahe_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="CLAHE: ", label_width=35, dropdown_width=40, value='FALSE', img='clahe', tooltip_key='simba2yolo_clahe')
+        self.single_id_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="REMOVE ANIMAL ID'S", label_width=35, dropdown_width=40, value='FALSE', img='mouse_head', tooltip_key='sleap_remove_animal_ids')
 
         settings_frm.grid(row=0, column=0, sticky=NW)
         self.sleap_dir .grid(row=0, column=0, sticky=NW)

simba/ui/pop_ups/video_processing_pop_up.py

@@ -51,7 +51,7 @@ from simba.utils.read_write import (
     concatenate_videos_in_folder, find_all_videos_in_directory, find_core_cnt,
     find_files_of_filetypes_in_directory, find_video_of_file, get_fn_ext,
     get_video_meta_data, seconds_to_timestamp, str_2_bool,
-    timestamp_to_seconds)
+    timestamp_to_seconds, read_frm_of_video)
 from simba.utils.warnings import FrameRangeWarning
 from simba.video_processors.brightness_contrast_ui import BrightnessContrastUI
 from simba.video_processors.clahe_ui import interactive_clahe_ui
@@ -1590,7 +1590,6 @@ class ClipMultipleVideosByFrameNumbersPopUp(PopUpMixin):
         check_if_dir_exists(in_dir=data_dir, source=self.__class__.__name__, create_if_not_exist=False )
         check_if_dir_exists(in_dir=save_dir, source=self.__class__.__name__, create_if_not_exist=True)
         self.video_paths = find_all_videos_in_directory(directory=data_dir, as_dict=True, raise_error=True)
-        print(self.video_paths)
         self.video_meta_data = [get_video_meta_data(video_path=x)["frame_count"]for x in list(self.video_paths.values())]
         max_video_name_len = len(max(list(self.video_paths.keys())))
         super().__init__(title="CLIP MULTIPLE VIDEOS BY FRAME NUMBERS", icon='clip')
@@ -1606,9 +1605,10 @@ class ClipMultipleVideosByFrameNumbersPopUp(PopUpMixin):
         seperator.grid(row=1, column=0, columnspan=5, rowspan=1, sticky="ew")
 
         self.entry_boxes, self.interactive_btns = {}, {}
-        for cnt, video_name in enumerate(self.video_paths.keys()):
+        for cnt, (video_name, video_path) in enumerate(self.video_paths.items()):
             self.entry_boxes[video_name] = {}
-            SimBALabel(parent=data_frm, font=Formats.FONT_REGULAR.value, txt=video_name + f' (frames: { self.video_meta_data[cnt]})', justify='left').grid(row=cnt + 2, column=0, padx=padx, sticky=NW)
+            img = read_frm_of_video(video_path=video_path, frame_index=0, size=(420, 280), keep_aspect_ratio=True, raise_error=False)
+            SimBALabel(parent=data_frm, font=Formats.FONT_REGULAR_BOLD.value, txt=video_name + f' (frames: { self.video_meta_data[cnt]})', justify='left', hover_img=img).grid(row=cnt + 2, column=0, padx=padx, sticky=NW)
             self.entry_boxes[video_name]["start"] = Entry_Box(data_frm, fileDescription="", labelwidth=0, validation="numeric", justify='center')
             self.entry_boxes[video_name]["end"] = Entry_Box(data_frm, fileDescription="", labelwidth=0, validation="numeric", justify='center')
             self.entry_boxes[video_name]["start"].grid(row=cnt + 2, column=2, sticky=NW, padx=padx)
@@ -1732,7 +1732,7 @@ class InitiateClipMultipleVideosByFrameNumbersPopUp(PopUpMixin):
             icon_link=Links.VIDEO_TOOLS.value,
         )
         self.input_folder = FolderSelect(
-            data_frm, "VIDEO DIRECTORY:", title="Select Folder with videos", lblwidth=30, lbl_icon='folder'
+            data_frm, "VIDEO DIRECTORY:", title="Select Folder with videos", lblwidth=30, lbl_icon='folder_video'
         )
         self.output_folder = FolderSelect(
             data_frm,
@@ -1814,9 +1814,10 @@ class ClipMultipleVideosByTimestamps(PopUpMixin):
         seperator.grid(row=1, column=0, columnspan=5, rowspan=1, sticky="ew")
 
         self.entry_boxes, self.interactive_btns = {}, {}
-        for cnt, video_name in enumerate(self.video_paths.keys()):
+        for cnt, (video_name, video_path) in enumerate(self.video_paths.items()):
             self.entry_boxes[video_name] = {}
-            SimBALabel(parent=data_frm, txt=video_name, justify='center').grid(row=cnt + 2, column=0, sticky=NW, padx=padx)
+            img = read_frm_of_video(video_path=video_path, frame_index=0, size=(420, 280), keep_aspect_ratio=True, raise_error=False)
+            SimBALabel(parent=data_frm, txt=video_name, justify='center', hover_img=img, font=Formats.FONT_REGULAR_BOLD.value).grid(row=cnt + 2, column=0, sticky=NW, padx=padx)
             video_length = self.video_meta_data[cnt]["video_length_s"]
             video_length_hhmmss = seconds_to_timestamp(seconds=video_length)
             SimBALabel(data_frm, txt=video_length_hhmmss, justify='center').grid(row=cnt + 2, column=1, sticky=NW, padx=padx)

simba/ui/pop_ups/yolo_plot_results.py

@@ -1,153 +1,146 @@
-from tkinter import *
-
-import numpy as np
-
-from simba.data_processors.cuda.utils import _is_cuda_available
-from simba.mixins.pop_up_mixin import PopUpMixin
-from simba.plotting.yolo_pose_track_visualizer import YOLOPoseTrackVisualizer
-from simba.plotting.yolo_pose_visualizer import YOLOPoseVisualizer
-from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, FileSelect,
-                                        FolderSelect, SimbaButton,
-                                        SimBADropDown)
-from simba.utils.checks import (check_file_exist_and_readable,
-                                check_if_dir_exists)
-from simba.utils.enums import Options, PackageNames
-from simba.utils.errors import (NoDataError, SimBAGPUError,
-                                SimBAPAckageVersionError)
-from simba.utils.printing import stdout_warning
-from simba.utils.read_write import (find_core_cnt,
-                                    find_files_of_filetypes_in_directory,
-                                    get_pkg_version, get_video_meta_data,
-                                    str_2_bool)
-from simba.utils.warnings import MissingFileWarning
-
-MAX_TRACKS_OPTIONS = ['None', 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
-CORE_CNT_OPTIONS = list(range(1, find_core_cnt()[0]))
-THRESHOLD_OPTIONS = list(np.arange(0.0, 1.1, 0.1).astype(np.float32))
-SIZE_OPTIONS = list(range(1, 21, 1))
-SIZE_OPTIONS.insert(0, 'AUTO')
-
-class YoloPoseVisualizerPopUp(PopUpMixin):
-
-    def __init__(self):
-        gpu_available, gpus = _is_cuda_available()
-        if not gpu_available:
-            raise SimBAGPUError(msg=f'Cannot train YOLO pose-estimation model. No NVIDA GPUs detected on machine', source=self.__class__.__name__)
-        ultralytics_version = get_pkg_version(pkg=PackageNames.ULTRALYTICS.value)
-        if ultralytics_version is None:
-            raise SimBAPAckageVersionError(msg=f'Cannot train YOLO pose-estimation model: Could not find ultralytics package in python environment',  source=self.__class__.__name__)
-
-        PopUpMixin.__init__(self, title="PLOT YOLO POSE ESTIMATION RESULTS", icon='ultralytics_2')
-        settings_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="SETTINGS", icon_name='settings')
-        self.save_dir = FolderSelect(settings_frm, folderDescription="SAVE DIRECTORY:", lblwidth=35, entry_width=45, lbl_icon='folder')
-        self.core_cnt_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=CORE_CNT_OPTIONS, label="CPU CORE COUNT:", label_width=35, dropdown_width=40, value=int(max(CORE_CNT_OPTIONS) / 3), img='cpu_small')
-        self.bbox_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="SHOW BOUNDING BOXES:",  label_width=35, dropdown_width=40, value='TRUE', img='rectangle_small')
-        self.verbose_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="VERBOSE:",  label_width=35, dropdown_width=40, value='TRUE', img='verbose')
-        self.threshold_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=THRESHOLD_OPTIONS, label="THRESHOLD:",  label_width=35, dropdown_width=40, value=0.0, img='threshold')
-        self.thickness_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=SIZE_OPTIONS, label="LINE THICKNESS:",  label_width=35, dropdown_width=40, value='AUTO', img='line')
-        self.circle_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=SIZE_OPTIONS, label="CIRCLE SIZE:", label_width=35, dropdown_width=40, value='AUTO', img='circle_small')
-        self.tracks_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="PLOT TRACKS:", label_width=35, dropdown_width=40, value='FALSE', img='path_2')
-
-        settings_frm.grid(row=0, column=0, sticky=NW)
-        self.save_dir.grid(row=0, column=0, sticky=NW)
-        self.tracks_dropdown.grid(row=1, column=0, sticky=NW)
-        self.core_cnt_dropdown.grid(row=2, column=0, sticky=NW)
-        self.bbox_dropdown.grid(row=3, column=0, sticky=NW)
-        self.verbose_dropdown.grid(row=4, column=0, sticky=NW)
-        self.threshold_dropdown.grid(row=5, column=0, sticky=NW)
-        self.thickness_dropdown.grid(row=6, column=0, sticky=NW)
-        self.circle_dropdown.grid(row=7, column=0, sticky=NW)
-
-        single_video_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="PLOT SINGLE VIDEO", icon_name='video')
-        self.data_path = FileSelect(parent=single_video_frm, fileDescription='DATA PATH (CSV):', lblwidth=35,  entry_width=45, file_types=[("YOLO CSV RESULT", ".csv")], lbl_icon='csv_grey')
-        self.video_path = FileSelect(parent=single_video_frm, fileDescription='VIDEO PATH:', lblwidth=35,  entry_width=45, file_types=[("YOLO CSV RESULT", Options.ALL_VIDEO_FORMAT_OPTIONS.value)], lbl_icon='video_2')
-        single_video_btn = SimbaButton(parent=single_video_frm, txt='CREATE SINGLE VIDEO', img='rocket', cmd=self.run, cmd_kwargs={'multiple': False})
-
-        single_video_frm.grid(row=1, column=0, sticky=NW)
-        self.data_path.grid(row=0, column=0, sticky=NW)
-        self.video_path.grid(row=1, column=0, sticky=NW)
-        single_video_btn.grid(row=2, column=0, sticky=NW)
-
-        multiple_video_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="PLOT MULTIPLE VIDEOS", icon_name='stack')
-        self.data_dir_path = FolderSelect(parent=multiple_video_frm, folderDescription='DATA DIRECTORY:', lblwidth=35,  entry_width=45, lbl_icon='folder')
-        self.video_dir_path = FolderSelect(parent=multiple_video_frm, folderDescription='VIDEO DIRECTORY:', lblwidth=35,  entry_width=45, lbl_icon='folder')
-        multiple_video_btn = SimbaButton(parent=multiple_video_frm, txt='CREATE MULTIPLE VIDEOS', img='rocket', cmd=self.run, cmd_kwargs={'multiple': True})
-
-        multiple_video_frm.grid(row=2, column=0, sticky=NW)
-        self.data_dir_path.grid(row=0, column=0, sticky=NW)
-        self.video_dir_path.grid(row=1, column=0, sticky=NW)
-        multiple_video_btn.grid(row=2, column=0, sticky=NW)
-
-        self.main_frm.mainloop()
-
-    def run(self, multiple: bool):
-
-        save_dir = self.save_dir.folder_path
-        core_cnt = int(self.core_cnt_dropdown.get_value())
-        bbox = str_2_bool(self.bbox_dropdown.get_value())
-        verbose = str_2_bool(self.verbose_dropdown.get_value())
-        threshold = float(self.threshold_dropdown.get_value())
-        thickness = None if self.thickness_dropdown.get_value() == 'AUTO' else int(self.threshold_dropdown.get_value())
-        circle_size = None if self.circle_dropdown.get_value() == 'AUTO' else int(self.circle_dropdown.get_value())
-        tracks = str_2_bool(self.tracks_dropdown.get_value())
-
-        Plotter = YOLOPoseTrackVisualizer if tracks else YOLOPoseVisualizer
-
-        if not multiple:
-            data_path = self.data_path.file_path
-            video_path = self.video_path.file_path
-            check_file_exist_and_readable(file_path=data_path, raise_error=True)
-            _ = get_video_meta_data(video_path=video_path)
-            plotter = Plotter(data_path=data_path,
-                              video_path=video_path,
-                              save_dir=save_dir,
-                              core_cnt=core_cnt,
-                              threshold=threshold,
-                              thickness=thickness,
-                              circle_size=circle_size,
-                              verbose=verbose,
-                              bbox=bbox)
-            plotter.run()
-        else:
-            data_dir = self.data_dir_path.folder_path
-            video_dir = self.video_dir_path.folder_path
-            check_if_dir_exists(in_dir=data_dir)
-            check_if_dir_exists(in_dir=video_dir)
-            data_paths = find_files_of_filetypes_in_directory(directory=data_dir, extensions=['.csv'], raise_error=True, as_dict=True)
-            video_paths = find_files_of_filetypes_in_directory(directory=video_dir, extensions=Options.ALL_VIDEO_FORMAT_OPTIONS.value, raise_error=True, as_dict=True)
-
-            missing_videos = list([x for x in video_paths.keys() if x not in data_paths.keys()])
-            missing_data_paths = list([x for x in data_paths.keys() if x not in video_paths.keys()])
-            if len(missing_videos) > 0:
-                MissingFileWarning(msg=f'Data files are missing video files in the {video_dir} directory: {missing_videos}', source=self.__class__.__name__)
-            if len(missing_data_paths) > 0:
-                MissingFileWarning(msg=f'Video files are missing data files in the {data_dir} directory: {missing_data_paths}', source=self.__class__.__name__)
-
-            data_paths = {k:v for k, v in data_paths.items() if k in video_paths.keys()}
-            if len(list(data_paths.keys())) == 0:
-                raise NoDataError(msg=f'No data file in the {data_dir} directory has a representative video file in the {video_dir} directory', source=self.__class__.__name__)
-            video_cnt = len(list(data_paths.keys()))
-            for cnt, (name, data_path) in enumerate(data_paths.items()):
-                if name in video_paths.keys():
-                    video_path = video_paths[name]
-                    print(f'Plotting YOLO results for video {name} (video {cnt+1}/{video_cnt}) ...')
-                    plotter = Plotter(data_path=data_path,
-                                      video_path=video_path,
-                                      save_dir=save_dir,
-                                      core_cnt=core_cnt,
-                                      threshold=threshold,
-                                      thickness=thickness,
-                                      circle_size=circle_size,
-                                      verbose=verbose,
-                                      bbox=bbox)
-                    plotter.run()
-                else:
-                    stdout_warning(msg=f'Skipping video {name} (no video exist in {video_dir})...')
-
-
-
-#YoloPoseVisualizerPopUp()
-
-
-
-
+from tkinter import *
+
+import numpy as np
+
+from simba.data_processors.cuda.utils import _is_cuda_available
+from simba.mixins.pop_up_mixin import PopUpMixin
+from simba.plotting.yolo_pose_track_visualizer import YOLOPoseTrackVisualizer
+from simba.plotting.yolo_pose_visualizer import YOLOPoseVisualizer
+from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, FileSelect,
+                                        FolderSelect, SimbaButton,
+                                        SimBADropDown)
+from simba.utils.checks import (check_file_exist_and_readable,
+                                check_if_dir_exists)
+from simba.utils.enums import Options, PackageNames
+from simba.utils.errors import (NoDataError, SimBAGPUError,
+                                SimBAPAckageVersionError)
+from simba.utils.printing import stdout_warning
+from simba.utils.read_write import (find_core_cnt,
+                                    find_files_of_filetypes_in_directory,
+                                    get_pkg_version, get_video_meta_data,
+                                    str_2_bool)
+from simba.utils.warnings import MissingFileWarning
+
+MAX_TRACKS_OPTIONS = ['None', 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
+CORE_CNT_OPTIONS = list(range(1, find_core_cnt()[0]))
+THRESHOLD_OPTIONS = list(np.arange(0.0, 1.1, 0.1).astype(np.float32))
+SIZE_OPTIONS = list(range(1, 21, 1))
+SIZE_OPTIONS.insert(0, 'AUTO')
+
+class YoloPoseVisualizerPopUp(PopUpMixin):
+
+    def __init__(self):
+        PopUpMixin.__init__(self, title="PLOT YOLO POSE ESTIMATION RESULTS", icon='ultralytics_2')
+        settings_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="SETTINGS", icon_name='settings')
+        self.save_dir = FolderSelect(settings_frm, folderDescription="SAVE DIRECTORY:", lblwidth=35, entry_width=45, lbl_icon='folder', tooltip_key='SAVE_DIR')
+        self.core_cnt_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=CORE_CNT_OPTIONS, label="CPU CORE COUNT:", label_width=35, dropdown_width=40, value=int(max(CORE_CNT_OPTIONS) / 3), img='cpu_small', tooltip_key='CORE_COUNT')
+        self.bbox_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="SHOW BOUNDING BOXES:",  label_width=35, dropdown_width=40, value='TRUE', img='rectangle_small', tooltip_key='SHOW_ANIMAL_BBOX')
+        self.verbose_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="VERBOSE:",  label_width=35, dropdown_width=40, value='TRUE', img='verbose', tooltip_key='verbose_dropdown')
+        self.threshold_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=THRESHOLD_OPTIONS, label="THRESHOLD:",  label_width=35, dropdown_width=40, value=0.0, img='threshold', tooltip_key='threshold_dropdown')
+        self.thickness_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=SIZE_OPTIONS, label="LINE THICKNESS:",  label_width=35, dropdown_width=40, value='AUTO', img='line', tooltip_key='yolo_plot_line_thickness')
+        self.circle_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=SIZE_OPTIONS, label="CIRCLE SIZE:", label_width=35, dropdown_width=40, value='AUTO', img='circle_small', tooltip_key='yolo_plot_circle_size')
+        self.tracks_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="PLOT TRACKS:", label_width=35, dropdown_width=40, value='FALSE', img='path_2', tooltip_key='yolo_plot_tracks')
+
+        settings_frm.grid(row=0, column=0, sticky=NW)
+        self.save_dir.grid(row=0, column=0, sticky=NW)
+        self.tracks_dropdown.grid(row=1, column=0, sticky=NW)
+        self.core_cnt_dropdown.grid(row=2, column=0, sticky=NW)
+        self.bbox_dropdown.grid(row=3, column=0, sticky=NW)
+        self.verbose_dropdown.grid(row=4, column=0, sticky=NW)
+        self.threshold_dropdown.grid(row=5, column=0, sticky=NW)
+        self.thickness_dropdown.grid(row=6, column=0, sticky=NW)
+        self.circle_dropdown.grid(row=7, column=0, sticky=NW)
+
+        single_video_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="PLOT SINGLE VIDEO", icon_name='video')
+        self.data_path = FileSelect(parent=single_video_frm, fileDescription='DATA PATH (CSV):', lblwidth=35,  entry_width=45, file_types=[("YOLO CSV RESULT", ".csv")], lbl_icon='csv_grey', tooltip_key='yolo_plot_data_path')
+        self.video_path = FileSelect(parent=single_video_frm, fileDescription='VIDEO PATH:', lblwidth=35,  entry_width=45, file_types=[("YOLO CSV RESULT", Options.ALL_VIDEO_FORMAT_OPTIONS.value)], lbl_icon='video_2', tooltip_key='yolo_plot_video_path')
+        single_video_btn = SimbaButton(parent=single_video_frm, txt='CREATE SINGLE VIDEO', img='rocket', cmd=self.run, cmd_kwargs={'multiple': False})
+
+        single_video_frm.grid(row=1, column=0, sticky=NW)
+        self.data_path.grid(row=0, column=0, sticky=NW)
+        self.video_path.grid(row=1, column=0, sticky=NW)
+        single_video_btn.grid(row=2, column=0, sticky=NW)
+
+        multiple_video_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="PLOT MULTIPLE VIDEOS", icon_name='stack')
+        self.data_dir_path = FolderSelect(parent=multiple_video_frm, folderDescription='DATA DIRECTORY:', lblwidth=35,  entry_width=45, lbl_icon='folder', tooltip_key='yolo_plot_data_dir')
+        self.video_dir_path = FolderSelect(parent=multiple_video_frm, folderDescription='VIDEO DIRECTORY:', lblwidth=35,  entry_width=45, lbl_icon='folder_video', tooltip_key='VIDEO_DIR')
+        multiple_video_btn = SimbaButton(parent=multiple_video_frm, txt='CREATE MULTIPLE VIDEOS', img='rocket', cmd=self.run, cmd_kwargs={'multiple': True})
+
+        multiple_video_frm.grid(row=2, column=0, sticky=NW)
+        self.data_dir_path.grid(row=0, column=0, sticky=NW)
+        self.video_dir_path.grid(row=1, column=0, sticky=NW)
+        multiple_video_btn.grid(row=2, column=0, sticky=NW)
+
+        self.main_frm.mainloop()
+
+    def run(self, multiple: bool):
+
+        save_dir = self.save_dir.folder_path
+        core_cnt = int(self.core_cnt_dropdown.get_value())
+        bbox = str_2_bool(self.bbox_dropdown.get_value())
+        verbose = str_2_bool(self.verbose_dropdown.get_value())
+        threshold = float(self.threshold_dropdown.get_value())
+        thickness = None if self.thickness_dropdown.get_value() == 'AUTO' else int(self.thickness_dropdown.get_value())
+        circle_size = None if self.circle_dropdown.get_value() == 'AUTO' else int(self.circle_dropdown.get_value())
+        tracks = str_2_bool(self.tracks_dropdown.get_value())
+
+        Plotter = YOLOPoseTrackVisualizer if tracks else YOLOPoseVisualizer
+
+        if not multiple:
+            data_path = self.data_path.file_path
+            video_path = self.video_path.file_path
+            check_file_exist_and_readable(file_path=data_path, raise_error=True)
+            _ = get_video_meta_data(video_path=video_path)
+            plotter = Plotter(data_path=data_path,
+                              video_path=video_path,
+                              save_dir=save_dir,
+                              core_cnt=core_cnt,
+                              threshold=threshold,
+                              thickness=thickness,
+                              circle_size=circle_size,
+                              verbose=verbose,
+                              bbox=bbox)
+            plotter.run()
+        else:
+            data_dir = self.data_dir_path.folder_path
+            video_dir = self.video_dir_path.folder_path
+            check_if_dir_exists(in_dir=data_dir)
+            check_if_dir_exists(in_dir=video_dir)
+            data_paths = find_files_of_filetypes_in_directory(directory=data_dir, extensions=['.csv'], raise_error=True, as_dict=True)
+            video_paths = find_files_of_filetypes_in_directory(directory=video_dir, extensions=Options.ALL_VIDEO_FORMAT_OPTIONS.value, raise_error=True, as_dict=True)
+
+            missing_videos = list([x for x in video_paths.keys() if x not in data_paths.keys()])
+            missing_data_paths = list([x for x in data_paths.keys() if x not in video_paths.keys()])
+            if len(missing_videos) > 0:
+                MissingFileWarning(msg=f'Data files are missing video files in the {video_dir} directory: {missing_videos}', source=self.__class__.__name__)
+            if len(missing_data_paths) > 0:
+                MissingFileWarning(msg=f'Video files are missing data files in the {data_dir} directory: {missing_data_paths}', source=self.__class__.__name__)
+
+            data_paths = {k:v for k, v in data_paths.items() if k in video_paths.keys()}
+            if len(list(data_paths.keys())) == 0:
+                raise NoDataError(msg=f'No data file in the {data_dir} directory has a representative video file in the {video_dir} directory', source=self.__class__.__name__)
+            video_cnt = len(list(data_paths.keys()))
+            for cnt, (name, data_path) in enumerate(data_paths.items()):
+                if name in video_paths.keys():
+                    video_path = video_paths[name]
+                    print(f'Plotting YOLO results for video {name} (video {cnt+1}/{video_cnt}) ...')
+                    plotter = Plotter(data_path=data_path,
+                                      video_path=video_path,
+                                      save_dir=save_dir,
+                                      core_cnt=core_cnt,
+                                      threshold=threshold,
+                                      thickness=thickness,
+                                      circle_size=circle_size,
+                                      verbose=verbose,
+                                      bbox=bbox)
+                    plotter.run()
+                else:
+                    stdout_warning(msg=f'Skipping video {name} (no video exist in {video_dir})...')
+
+
+
+#YoloPoseVisualizerPopUp()
+
+
+
+

simba/ui/pop_ups/yolo_pose_train_popup.py

@@ -1,9 +1,12 @@
-
+import os
+import subprocess
+import sys
+import tempfile
 from tkinter import *
+from tkinter import messagebox
 
 from simba.data_processors.cuda.utils import _is_cuda_available
 from simba.mixins.pop_up_mixin import PopUpMixin
-from simba.model.yolo_fit import FitYolo
 from simba.third_party_label_appenders.transform.utils import \
     check_valid_yolo_map
 from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, FileSelect,
@@ -17,10 +20,13 @@ from simba.utils.read_write import find_core_cnt, get_pkg_version, str_2_bool
 EPOCH_OPTIONS = list(range(100, 5750, 250))
 PATIENCE_OPTIONS = list(range(50, 1050, 50))
 IMG_SIZE_OPTIONS = [256, 320, 416, 480, 512, 640, 720, 768, 960, 1280]
-CORE_CNT_OPTIONS = list(range(1, find_core_cnt()[0]))
+# On Windows, PyTorch DataLoader often deadlocks with workers > 8; cap options to avoid hang
+_max_workers = min(find_core_cnt()[0], 8) if sys.platform == 'win32' else find_core_cnt()[0]
+CORE_CNT_OPTIONS = list(range(1, _max_workers + 1))
 BATCH_SIZE_OPTIONS =  [2, 4, 8, 16, 32, 64, 128]
 devices = ['CPU']
-
+FORMAT_OPTIONS =  Options.VALID_YOLO_FORMATS.value
+FORMAT_OPTIONS.insert(0, 'None')
 class YOLOPoseTrainPopUP(PopUpMixin):
     def __init__(self):
         gpu_available, gpus = _is_cuda_available()
@@ -33,19 +39,19 @@ class YOLOPoseTrainPopUP(PopUpMixin):
         PopUpMixin.__init__(self, title="TRAIN YOLO POSE ESTIMATION MODEL", icon='ultralytics_2')
         settings_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="SETTINGS", icon_name='settings')
         devices.extend([f'{x} : {y["model"]}' for x, y in gpus.items()])
-        self.yolo_map_path = FileSelect(parent=settings_frm, fileDescription='YOLO MAP FILE (YAML):', lblwidth=35, entry_width=45, file_types=[("YOLO MODEL FILE", ".yaml")], lbl_icon='file')
-        self.save_dir = FolderSelect(settings_frm, folderDescription="SAVE DIRECTORY:", lblwidth=35, entry_width=45, lbl_icon='save')
-        self.weights_path = FileSelect(parent=settings_frm, fileDescription='INITIAL WEIGHT FILE (E.G., .PT):', lblwidth=35, entry_width=45, lbl_icon='file')
+        self.yolo_map_path = FileSelect(parent=settings_frm, fileDescription='YOLO MAP FILE (YAML):', lblwidth=35, entry_width=45, file_types=[("YOLO MODEL FILE", ".yaml")], lbl_icon='file', tooltip_key='yolo_map_path')
+        self.save_dir = FolderSelect(settings_frm, folderDescription="SAVE DIRECTORY:", lblwidth=35, entry_width=45, lbl_icon='save', tooltip_key='SAVE_DIR')
+        self.weights_path = FileSelect(parent=settings_frm, fileDescription='INITIAL WEIGHT FILE (E.G., .PT):', lblwidth=35, entry_width=45, lbl_icon='file', tooltip_key='yolo_initial_weights_path')
 
-        self.epochs_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=EPOCH_OPTIONS, label="EPOCHS: ", label_width=35, dropdown_width=40, value=500, img='rotate')
-        self.batch_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=BATCH_SIZE_OPTIONS, label="BATCH SIZE: ", label_width=35, dropdown_width=40, value=16, img='weight')
-        self.plots_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="PLOTS:", label_width=35, dropdown_width=40, value='TRUE', img='plot')
-        self.verbose_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="VERBOSE:", label_width=35, dropdown_width=40, value='TRUE', img='verbose')
-        self.workers_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=CORE_CNT_OPTIONS, label="CPU WORKERS:", label_width=35, dropdown_width=40, value=int(max(CORE_CNT_OPTIONS)/2), img='cpu_small')
-        self.format_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=Options.VALID_YOLO_FORMATS.value, label="FORMAT:", label_width=35, dropdown_width=40, value='engine', img='file_type')
-        self.img_size_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=IMG_SIZE_OPTIONS, label="IMAGE SIZE:", label_width=35, dropdown_width=40, value=640, img='resize')
-        self.patience_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=PATIENCE_OPTIONS, label="PATIENCE:", label_width=35, dropdown_width=40, value=100, img='timer')
-        self.devices_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=devices, label="DEVICE:", label_width=35, dropdown_width=40, value=devices[1], img='gpu_3')
+        self.epochs_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=EPOCH_OPTIONS, label="EPOCHS: ", label_width=35, dropdown_width=40, value=500, img='rotate', tooltip_key='epochs_dropdown')
+        self.batch_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=BATCH_SIZE_OPTIONS, label="BATCH SIZE: ", label_width=35, dropdown_width=40, value=16, img='weight', tooltip_key='batch_dropdown')
+        self.plots_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="PLOTS:", label_width=35, dropdown_width=40, value='TRUE', img='plot', tooltip_key='plots_dropdown')
+        self.verbose_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label="VERBOSE:", label_width=35, dropdown_width=40, value='TRUE', img='verbose', tooltip_key='verbose_dropdown')
+        self.workers_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=CORE_CNT_OPTIONS, label="CPU WORKERS:", label_width=35, dropdown_width=40, value=int(max(CORE_CNT_OPTIONS)/2), img='cpu_small', tooltip_key='workers_dropdown')
+        self.format_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=FORMAT_OPTIONS, label="FORMAT:", label_width=35, dropdown_width=40, value='None', img='file_type', tooltip_key='format_dropdown')
+        self.img_size_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=IMG_SIZE_OPTIONS, label="IMAGE SIZE:", label_width=35, dropdown_width=40, value=640, img='resize', tooltip_key='img_size_dropdown')
+        self.patience_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=PATIENCE_OPTIONS, label="PATIENCE:", label_width=35, dropdown_width=40, value=100, img='timer', tooltip_key='patience_dropdown')
+        self.devices_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=devices, label="DEVICE:", label_width=35, dropdown_width=40, value=devices[1], img='gpu_3', tooltip_key='devices_dropdown')
 
         settings_frm.grid(row=0, column=0, sticky=NW)
         self.yolo_map_path.grid(row=0, column=0, sticky=NW)
@@ -74,16 +80,56 @@ class YOLOPoseTrainPopUP(PopUpMixin):
         workers = int(self.workers_dropdown.get_value())
         batch_size = int(self.batch_dropdown.get_value())
         device = self.devices_dropdown.get_value()
-        device = 'cpu' if device == 'CPU' else int(device.split(':', 1)[0])
-        format = self.format_dropdown.get_value()
+        device_str = 'cpu' if device == 'CPU' else device.split(':', 1)[0]
+        format_val = None if self.format_dropdown.get_value() == 'None' else self.format_dropdown.get_value()
         imgsz = int(self.img_size_dropdown.get_value())
         patience = int(self.patience_dropdown.get_value())
-
         check_if_dir_exists(in_dir=save_dir, source=f'{self.__class__.__name__} SAVE DIRECTORY')
-        check_file_exist_and_readable(file_path=weights_path, raise_error=True)
+        if not check_file_exist_and_readable(file_path=weights_path, raise_error=False):
+            weights_path = None
         check_file_exist_and_readable(file_path=yolo_map_path, raise_error=True)
         check_valid_yolo_map(yolo_map=yolo_map_path)
-        runner = FitYolo(weights_path=weights_path, model_yaml=yolo_map_path, save_path=save_dir, epochs=epochs, batch=batch_size, plots=plots, format=format, device=device, verbose=verbose, workers=workers, imgsz=imgsz, patience=patience)
-        runner.run()
+        workers_for_subprocess = min(workers, 8) if sys.platform == 'win32' else workers
+        cmd = [
+            sys.executable, '-m', 'simba.model.yolo_fit',
+            '--model_yaml', yolo_map_path,
+            '--save_path', save_dir,
+            '--epochs', str(epochs),
+            '--batch', str(batch_size),
+            '--plots', 'True' if plots else 'False',
+            '--imgsz', str(imgsz),
+            '--device', str(device_str),
+            '--verbose', 'True' if verbose else 'False',
+            '--workers', str(workers_for_subprocess),
+            '--patience', str(patience),
+        ]
+        if weights_path is not None:
+            cmd.extend(['--weights_path', weights_path])
+        if format_val is not None:
+            cmd.extend(['--format', format_val])
+
+        creationflags = subprocess.CREATE_NEW_CONSOLE if sys.platform == 'win32' else 0
+        env = os.environ.copy()
+        env['MPLBACKEND'] = 'Agg'
+        try:
+            if sys.platform == 'win32':
+                cmd_line = subprocess.list2cmdline(cmd)
+                with tempfile.NamedTemporaryFile(mode='w', suffix='.bat', delete=False, newline='') as f:
+                    f.write('@echo off\n' + cmd_line + '\npause\n')
+                    bat_path = f.name
+                subprocess.Popen([bat_path], creationflags=creationflags, env=env)
+            else:
+                subprocess.Popen(cmd, creationflags=creationflags, env=env)
+        except Exception as e:
+            messagebox.showerror('YOLO training', f'Failed to start training process: {e}')
+            return
+        msg = (
+            'YOLO training has been started in a separate process to avoid memory issues.\n\n'
+            'On Windows a new console window will show training progress. '
+            'On other platforms, check the terminal from which SimBA was launched.\n\n'
+            f'Results will be saved to:\n{save_dir}'
+        )
+        messagebox.showinfo('YOLO training started', msg)
+
 
-#YOLOPoseTrainPopUP()
+#@YOLOPoseTrainPopUP()