diff --git a/Start.bat b/Start.bat index 139d668d..b588bea5 100644 --- a/Start.bat +++ b/Start.bat @@ -1,5 +1,6 @@ - -call conda activate visomaster +@echo off +call "%USERPROFILE%\miniconda3\condabin\conda.bat" activate visomaster +SET KMP_DUPLICATE_LIB_OK=TRUE call app/ui/core/convert_ui_to_py.bat SET APP_ROOT=%~dp0 SET APP_ROOT=%APP_ROOT:~0,-1% diff --git a/app/processors/models_processor.py b/app/processors/models_processor.py index 070c4c01..5eba0cb3 100644 --- a/app/processors/models_processor.py +++ b/app/processors/models_processor.py @@ -126,8 +126,8 @@ def load_model(self, model_name, session_options=None): with self.model_lock: self.main_window.model_loading_signal.emit() # QApplication.processEvents() - # if not is_file_exists(self.models_path[model_name]): - # download_file(model_name, self.models_path[model_name], self.models_data[model_name]['hash'], self.models_data[model_name]['url']) + if not is_file_exists(self.models_path[model_name]): + download_file(model_name, self.models_path[model_name], self.models_data[model_name]['hash'], self.models_data[model_name]['url']) if session_options is None: model_instance = onnxruntime.InferenceSession(self.models_path[model_name], providers=self.providers) else: diff --git a/app/ui/core/media_rc.py b/app/ui/core/media_rc.py index 249fd4d3..51255889 100644 --- a/app/ui/core/media_rc.py +++ b/app/ui/core/media_rc.py @@ -292458,95 +292458,95 @@ \x00\x00\x00\x00\x00\x02\x00\x00\x00-\x00\x00\x00\x03\ \x00\x00\x00\x00\x00\x00\x00\x00\ \x00\x00\x04\xc2\x00\x00\x00\x00\x00\x01\x00\x19\x1f\xed\ -\x00\x00\x01\x94\xc0Z)Q\ +\x00\x00\x01\x9c\x04\xe6\xd6\x8d\ \x00\x00\x01\x9a\x00\x00\x00\x00\x00\x01\x00\x02\x8f\xdc\ -\x00\x00\x01\x94\xc0Z)N\ +\x00\x00\x01\x9c\x04\xe6\xd6\x89\ \x00\x00\x00\xf8\x00\x00\x00\x00\x00\x01\x00\x01\x11\x9d\ -\x00\x00\x01\x94\xc0Z)`\ +\x00\x00\x01\x9c\x04\xe6\xd6\x9f\ \x00\x00\x02p\x00\x00\x00\x00\x00\x01\x00\x02\xe9\xb3\ -\x00\x00\x01\x94\xc0Z)L\ +\x00\x00\x01\x9c\x04\xe6\xd6\x86\ \x00\x00\x02@\x00\x00\x00\x00\x00\x01\x00\x02\xd0\x0d\ -\x00\x00\x01\x94\xc0Z)L\ +\x00\x00\x01\x9c\x04\xe6\xd6\x87\ \x00\x00\x03\xe0\x00\x00\x00\x00\x00\x01\x00\x18B\xa7\ -\x00\x00\x01\x94\xc0Z)G\ +\x00\x00\x01\x9c\x04\xe6\xd6\x82\ \x00\x00\x01\xfa\x00\x00\x00\x00\x00\x01\x00\x02\xb6\xdd\ -\x00\x00\x01\x94\xc0Z)a\ +\x00\x00\x01\x9c\x04\xe6\xd6\xa1\ \x00\x00\x02\xc8\x00\x00\x00\x00\x00\x01\x00\x03\x0c8\ -\x00\x00\x01\x94\xc0Z)R\ +\x00\x00\x01\x9c\x04\xe6\xd6\x8f\ \x00\x00\x04\x94\x00\x00\x00\x00\x00\x01\x00\x18\xe0\xb4\ -\x00\x00\x01\x94\xc0Z)g\ +\x00\x00\x01\x9c\x04\xe6\xd6\xa6\ \x00\x00\x00\xb6\x00\x00\x00\x00\x00\x01\x00\x00\x84'\ -\x00\x00\x01\x94\xc0Z)K\ +\x00\x00\x01\x9c\x04\xe6\xd6\x85\ \x00\x00\x00\xd8\x00\x00\x00\x00\x00\x01\x00\x01\x04\xac\ -\x00\x00\x01\x94\xc0Z)`\ +\x00\x00\x01\x9c\x04\xe6\xd6\xa0\ \x00\x00\x04N\x00\x00\x00\x00\x00\x01\x00\x18\xc3\xf4\ -\x00\x00\x01\x94\xc0Z)a\ +\x00\x00\x01\x9c\x04\xe6\xd6\xa1\ \x00\x00\x01H\x00\x00\x00\x00\x00\x01\x00\x01*V\ -\x00\x00\x01\x94\xc0Z)I\ +\x00\x00\x01\x9c\x04\xe6\xd6\x84\ \x00\x00\x05\xfa\x00\x00\x00\x00\x00\x01\x00G\x0c\xdc\ -\x00\x00\x01\x94\xc0Z)M\ +\x00\x00\x01\x9c\x04\xe6\xd6\x88\ \x00\x00\x00.\x00\x00\x00\x00\x00\x01\x00\x00\x03\xb8\ -\x00\x00\x01\x94\xc0Z)O\ +\x00\x00\x01\x9c\x04\xe6\xd6\x8a\ \x00\x00\x03\x22\x00\x00\x00\x00\x00\x01\x00\x17\xdaL\ -\x00\x00\x01\x94\xc0Z)P\ +\x00\x00\x01\x9c\x04\xe6\xd6\x8c\ \x00\x00\x01\x1a\x00\x00\x00\x00\x00\x01\x00\x01\x1f)\ -\x00\x00\x01\x94\xc0Z)H\ +\x00\x00\x01\x9c\x04\xe6\xd6\x83\ \x00\x00\x02\xee\x00\x00\x00\x00\x00\x01\x00\x03?\x18\ -\x00\x00\x01\x94\xc0Z)O\ +\x00\x00\x01\x9c\x04\xe6\xd6\x8b\ \x00\x00\x02\x8a\x00\x00\x00\x00\x00\x01\x00\x02\xec\xf8\ -\x00\x00\x01\x94\xc0Z)N\ +\x00\x00\x01\x9c\x04\xe6\xd6\x89\ \x00\x00\x03\x9e\x00\x00\x00\x00\x00\x01\x00\x181[\ -\x00\x00\x01\x94\xc0Z)_\ +\x00\x00\x01\x9c\x04\xe6\xd6\x9e\ \x00\x00\x00J\x00\x00\x00\x00\x00\x01\x00\x00\x11\xd2\ -\x00\x00\x01\x94\xc0Z)N\ +\x00\x00\x01\x9c\x04\xe6\xd6\x8a\ \x00\x00\x04\x22\x00\x00\x00\x00\x00\x01\x00\x18\xb7`\ -\x00\x00\x01\x94\xc0Z)M\ +\x00\x00\x01\x9c\x04\xe6\xd6\x87\ \x00\x00\x05\x9c\x00\x00\x00\x00\x00\x01\x00F\xa6\xfc\ -\x00\x00\x01\x94\xc0Z)K\ +\x00\x00\x01\x9c\x04\xe6\xd6\x86\ \x00\x00\x01h\x00\x00\x00\x00\x00\x01\x00\x01O\xda\ -\x00\x00\x01\x94\xc0Z)H\ +\x00\x00\x01\x9c\x04\xe6\xd6\x83\ \x00\x00\x03\x08\x00\x00\x00\x00\x00\x01\x00\x03m\xe6\ -\x00\x00\x01\x94\xc0Z)V\ +\x00\x00\x01\x9c\x04\xe6\xd6\x93\ \x00\x00\x01\x84\x00\x00\x00\x00\x00\x01\x00\x02\x88\x0f\ -\x00\x00\x01\x94\xc0Z)Q\ +\x00\x00\x01\x9c\x04\xe6\xd6\x8e\ \x00\x00\x05\x0e\x00\x00\x00\x00\x00\x01\x00E\xd7\xab\ -\x00\x00\x01\x94\xc0Z)h\ +\x00\x00\x01\x9c\x04\xe6\xd6\xa7\ \x00\x00\x03\xfe\x00\x00\x00\x00\x00\x01\x00\x18\xa7\xef\ -\x00\x00\x01\x94\xc0Z)a\ +\x00\x00\x01\x9c\x04\xe6\xd6\xa0\ \x00\x00\x02\xa6\x00\x00\x00\x00\x00\x01\x00\x02\xfe\xe8\ -\x00\x00\x01\x94\xc0Z)M\ +\x00\x00\x01\x9c\x04\xe6\xd6\x88\ \x00\x00\x04\xea\x00\x00\x00\x00\x00\x01\x00\x19\xab\xdf\ -\x00\x00\x01\x94\xc0Z)^\ +\x00\x00\x01\x9c\x04\xe6\xd6\x9d\ \x00\x00\x05r\x00\x00\x00\x00\x00\x01\x00F\x98\x08\ -\x00\x00\x01\x94\xc0Z)a\ +\x00\x00\x01\x9c\x04\xe6\xd6\xa1\ \x00\x00\x05\xd8\x00\x00\x00\x00\x00\x01\x00G\x09.\ -\x00\x00\x01\x94\xc0Z)_\ +\x00\x00\x01\x9c\x04\xe6\xd6\x9d\ \x00\x00\x00\x10\x00\x00\x00\x00\x00\x01\x00\x00\x00\x00\ -\x00\x00\x01\x94\xc0Z)_\ +\x00\x00\x01\x9c\x04\xe6\xd6\x9e\ \x00\x00\x01\xd2\x00\x00\x00\x00\x00\x01\x00\x02\xa7n\ -\x00\x00\x01\x94\xc0Z)`\ +\x00\x00\x01\x9c\x04\xe6\xd6\xa0\ \x00\x00\x03\x86\x00\x00\x00\x00\x00\x01\x00\x17\xe3\xca\ -\x00\x00\x01\x94\xc0Z)c\ +\x00\x00\x01\x9c\x04\xe6\xd6\xa2\ \x00\x00\x05\xb4\x00\x00\x00\x00\x00\x01\x00F\xfdK\ -\x00\x00\x01\x94\xc0Z)L\ +\x00\x00\x01\x9c\x04\xe6\xd6\x86\ \x00\x00\x00\x98\x00\x00\x00\x00\x00\x01\x00\x00w\x89\ -\x00\x00\x01\x94\xc0Z)M\ +\x00\x00\x01\x9c\x04\xe6\xd6\x89\ \x00\x00\x06\x1a\x00\x00\x00\x00\x00\x01\x00G2\xf7\ -\x00\x00\x01\x94\xc0Z)I\ +\x00\x00\x01\x9c\x04\xe6\xd6\x84\ \x00\x00\x02\x1c\x00\x00\x00\x00\x00\x01\x00\x02\xc3\x9e\ -\x00\x00\x01\x94\xc0Z)a\ +\x00\x00\x01\x9c\x04\xe6\xd6\xa2\ \x00\x00\x00~\x00\x00\x00\x00\x00\x01\x00\x00\x1f#\ -\x00\x00\x01\x94\xc0Z)P\ +\x00\x00\x01\x9c\x04\xe6\xd6\x8c\ \x00\x00\x05H\x00\x00\x00\x00\x00\x01\x00F\x8c\xf1\ -\x00\x00\x01\x94\xc0Z)I\ +\x00\x00\x01\x9c\x04\xe6\xd6\x84\ \x00\x00\x03\xba\x00\x00\x00\x00\x00\x01\x00\x185\x09\ -\x00\x00\x01\x94\xc0Z)_\ +\x00\x00\x01\x9c\x04\xe6\xd6\x9f\ \x00\x00\x04t\x00\x00\x00\x00\x00\x01\x00\x18\xd2\xe0\ -\x00\x00\x01\x94\xc0Z)O\ +\x00\x00\x01\x9c\x04\xe6\xd6\x8a\ \x00\x00\x03R\x00\x00\x00\x00\x00\x01\x00\x17\xdf\x0b\ -\x00\x00\x01\x94\xc0Z)P\ +\x00\x00\x01\x9c\x04\xe6\xd6\x8b\ \x00\x00\x05(\x00\x00\x00\x00\x00\x01\x00Fo\x09\ -\x00\x00\x01\x94\xc0Z)Q\ +\x00\x00\x01\x9c\x04\xe6\xd6\x8e\ " def qInitResources(): diff --git a/app/ui/main_ui.py b/app/ui/main_ui.py index 8aff4341..77f3a882 100644 --- a/app/ui/main_ui.py +++ b/app/ui/main_ui.py @@ -107,9 +107,9 @@ def initialize_widgets(self): # Set up click to select and drop action on ListWidgets self.targetVideosList.setAcceptDrops(True) - self.targetVideosList.viewport().setAcceptDrops(False) + self.targetVideosList.viewport().setAcceptDrops(True) self.inputFacesList.setAcceptDrops(True) - self.inputFacesList.viewport().setAcceptDrops(False) + self.inputFacesList.viewport().setAcceptDrops(True) list_widget_event_filter = ListWidgetEventFilter(self, self) self.targetVideosList.installEventFilter(list_widget_event_filter) self.targetVideosList.viewport().installEventFilter(list_widget_event_filter) diff --git a/app/ui/widgets/ui_workers.py b/app/ui/widgets/ui_workers.py index b1144247..63ea5d39 100644 --- a/app/ui/widgets/ui_workers.py +++ b/app/ui/widgets/ui_workers.py @@ -54,10 +54,9 @@ def load_videos_and_images_from_folder(self, folder_name): i=0 media_files = video_files + image_files - for media_file in media_files: + for media_file_path in media_files: if not self._running: # Check if the thread is still running break - media_file_path = os.path.join(folder_name, media_file) file_type = misc_helpers.get_file_type(media_file_path) pixmap = common_widget_actions.extract_frame_as_pixmap(media_file_path, file_type) if self.media_ids: @@ -166,8 +165,6 @@ def load_faces(self, folder_name=False, files_list=None): break if not misc_helpers.is_image_file(image_file_path): return - if folder_name: - image_file_path = os.path.join(folder_name, image_file_path) frame = misc_helpers.read_image_file(image_file_path) if frame is None: continue