From 7663076e92fa8e137720e7a922f30c12636ecb41 Mon Sep 17 00:00:00 2001 From: BlueDarkUP <119296966+BlueDarkUP@users.noreply.github.com> Date: Sun, 24 May 2026 22:08:57 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E4=BA=86=E8=AE=B8=E5=A4=9A?= =?UTF-8?q?=E4=BD=BF=E7=94=A8=E4=B8=8A=E7=9A=84bug=20app.py:=201.=E5=8E=9F?= =?UTF-8?q?=E6=9D=A5=E5=9C=A8app.py=E4=B8=AD=E3=80=82=E8=AE=BE=E7=BD=AE?= =?UTF-8?q?=E4=BA=86sys.stdout=20=3D=20io.TextIOWrapper(sys.stdout.buffer,?= =?UTF-8?q?=20encoding=3D'utf-8')=EF=BC=8C=E4=B8=BA=E4=BA=86=E5=8F=8A?= =?UTF-8?q?=E6=97=B6=E6=8D=95=E8=8E=B7=E6=8A=A5=E9=94=99=E6=B7=BB=E5=8A=A0?= =?UTF-8?q?=E4=BA=86write=5Fthrough=3DTrue=202.=E5=AF=B9=E4=BA=8E=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=E7=9A=84=E5=AE=89=E8=A3=85=EF=BC=8C=E5=8F=91=E7=8E=B0?= =?UTF-8?q?=E9=97=AE=E9=A2=98=EF=BC=9A=E5=BD=93=E8=AE=BE=E7=BD=AE=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=E4=B8=BAyolo11n=EF=BC=8C=E4=BC=9A=E6=98=BE=E7=A4=BA?= =?UTF-8?q?=E5=AE=89=E8=A3=85=E4=BD=86=E6=98=AF=E6=97=A0=E6=B3=95=E5=AE=89?= =?UTF-8?q?=E8=A3=85=E6=88=90=E5=8A=9F=EF=BC=8C=E8=BF=99=E6=98=AF=E4=B8=80?= =?UTF-8?q?=E4=B8=AA=E5=9B=BA=E4=BB=B6bug=EF=BC=8C=E8=80=83=E8=99=91?= =?UTF-8?q?=E5=88=B0YOLO11=E5=92=8CYOLOv8=20=E5=9C=A8=E7=9B=AE=E6=A0=87?= =?UTF-8?q?=E6=A3=80=E6=B5=8B=E4=B8=8A=E7=9A=84=E7=BD=91=E7=BB=9C=E8=BE=93?= =?UTF-8?q?=E5=87=BA=E7=BB=93=E6=9E=84=E3=80=81=E5=90=8E=E5=A4=84=E7=90=86?= =?UTF-8?q?=E7=AE=97=E6=B3=95=E9=83=BD=E5=AE=8C=E5=85=A8=E7=9B=B8=E5=90=8C?= =?UTF-8?q?=EF=BC=8C=E6=89=80=E4=BB=A585-93=E8=A1=8C=E5=B0=86=E6=89=80?= =?UTF-8?q?=E6=9C=89yolo11=E5=85=A8=E9=83=A8=E5=AF=B9=E5=BA=94=E4=B8=BAyol?= =?UTF-8?q?ov8n=EF=BC=88=E5=BB=BA=E8=AE=AE=E4=B9=8B=E5=90=8E=E4=BF=AE?= =?UTF-8?q?=E5=A4=8D=E8=BF=99=E4=B8=AA=E5=9B=BA=E4=BB=B6=E9=97=AE=E9=A2=98?= =?UTF-8?q?=EF=BC=89=203.=E7=94=B1=E4=BA=8Ec=E5=90=8E=E7=AB=AF=E4=BC=9A?= =?UTF-8?q?=E7=94=B1=E4=BA=8E=E6=97=A0=E6=B3=95=E8=A7=A3=E6=9E=90python?= =?UTF-8?q?=E6=8A=A5=E9=94=99=E7=9B=B4=E6=8E=A5abord=EF=BC=8C=E6=89=80?= =?UTF-8?q?=E4=BB=A5=E6=B7=BB=E5=8A=A0=E4=BA=86=E5=BC=82=E5=B8=B8=E5=A4=84?= =?UTF-8?q?=E7=90=86=204.PyQt=E7=9A=84=E4=B8=BB=E7=BA=BF=E7=A8=8B=E5=BC=82?= =?UTF-8?q?=E5=B8=B8=E5=AF=BC=E8=87=B40xC0000409=E5=BC=BA=E5=88=B6?= =?UTF-8?q?=E9=80=80=E5=87=BA=EF=BC=8C=E6=97=A0=E6=B3=95=E8=8E=B7=E5=8F=96?= =?UTF-8?q?=E5=86=85=E9=83=A8=E6=8A=A5=E9=94=99=EF=BC=8C=E5=9C=A8=E4=B8=BB?= =?UTF-8?q?=E7=A8=8B=E5=BA=8F=E4=B8=AD=E6=B7=BB=E5=8A=A0=E4=BB=A3=E7=A0=81?= =?UTF-8?q?=E6=8B=A6=E6=88=AA=E4=B8=BB=E7=BA=BF=E7=A8=8B=E5=BC=82=E5=B8=B8?= =?UTF-8?q?=20confertor.py:=201.nncase=E7=9A=84import=E6=97=B6=E9=97=B4?= =?UTF-8?q?=E4=B8=8D=E6=81=B0=E5=BD=93=EF=BC=8C=E5=8E=9F=E6=9D=A5nncase?= =?UTF-8?q?=E5=9C=A8=E7=8E=AF=E5=A2=83=E5=8F=98=E9=87=8F=E8=AE=BE=E7=BD=AE?= =?UTF-8?q?=E5=89=8D=E5=B0=B1=E5=88=9D=E5=A7=8B=E5=8C=96=EF=BC=8C=E5=AF=BC?= =?UTF-8?q?=E8=87=B4nncase=E6=97=A0=E6=B3=95=E4=BD=BF=E7=94=A8=E5=8E=9F?= =?UTF-8?q?=E6=9C=AC=E5=9C=A8=E5=85=B6=E5=88=9D=E5=A7=8B=E5=8C=96=E5=90=8E?= =?UTF-8?q?=E6=89=8D=E8=AE=BE=E7=BD=AE=E7=9A=84=E7=8E=AF=E5=A2=83=E5=8F=98?= =?UTF-8?q?=E9=87=8F=202.=E5=A4=A7=E5=9E=8B=E6=95=B0=E6=8D=AE=E9=9B=86?= =?UTF-8?q?=E8=BF=9B=E8=A1=8C=E6=A0=A1=E5=87=86=E6=95=B0=E6=8D=AE=E9=87=8F?= =?UTF-8?q?=E5=B7=A8=E5=A4=A7=EF=BC=8C=E5=9C=A8gen=E5=87=BD=E6=95=B0?= =?UTF-8?q?=E4=B8=AD=E6=B7=BB=E5=8A=A0=E4=B8=8A=E9=99=90=E4=B8=BA100?= =?UTF-8?q?=EF=BC=8C=E5=9B=A0=E4=B8=BA=E6=A0=A1=E5=87=86=E5=B9=B6=E4=B8=8D?= =?UTF-8?q?=E9=9C=80=E8=A6=81=E5=A4=A7=E9=87=8F=E7=9A=84=E6=95=B0=E6=8D=AE?= =?UTF-8?q?=203.=E5=A6=82=E6=9E=9Cyaml=E4=B8=AD=E5=8C=85=E5=90=AB./?= =?UTF-8?q?=EF=BC=8C=E4=BC=9A=E5=AF=BC=E8=87=B4=E7=9B=AE=E5=BD=95=E2=80=9C?= =?UTF-8?q?.=E2=80=9D=E8=A2=AB=E9=94=99=E8=AF=AF=E8=AF=BB=E5=8F=96?= =?UTF-8?q?=E5=AF=BC=E8=87=B4=E6=A0=A1=E5=87=86=E7=85=A7=E7=89=87=E6=97=A0?= =?UTF-8?q?=E6=B3=95=E8=A2=AB=E6=88=90=E5=8A=9F=E8=AF=BB=E5=8F=96=EF=BC=8C?= =?UTF-8?q?=E8=BF=9B=E8=A1=8C=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app.py | 43 ++++++++++++++++++++++++++++++------------- convertor.py | 50 +++++++++++++++++++++++++------------------------- 2 files changed, 55 insertions(+), 38 deletions(-) diff --git a/app.py b/app.py index 9c2c7fa..7376c3b 100644 --- a/app.py +++ b/app.py @@ -22,7 +22,7 @@ from PyQt5.QtGui import QPixmap from PyQt5.QtCore import QThread, pyqtSignal import io -sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8') +sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8', write_through=True) lang_id = 0 lang = { @@ -82,9 +82,15 @@ "yolov8n-cls": ["yolov8n-cls", "object-classification-classifier"], "yolov8n-seg": ["yolov8n-seg", "object-segmentation-segment"], "yolov8n": ["yolov8n-det", "object-detection-detector"], - "yolo11n-cls": ["yolo11n-cls", "object-classification-classifier"], - "yolo11n-seg": ["yolo11n-seg", "object-segmentation-segment"], - "yolo11n": ["yolo11n-det", "object-detection-detector"], + + "yolo11n-cls": ["yolov8n-cls", "object-classification-classifier"], + "yolov11n-cls": ["yolov8n-cls", "object-classification-classifier"], + + "yolo11n-seg": ["yolov8n-seg", "object-segmentation-segment"], + "yolov11n-seg": ["yolov8n-seg", "object-segmentation-segment"], + + "yolo11n": ["yolov8n-det", "object-detection-detector"], + "yolov11n": ["yolov8n-det", "object-detection-detector"], } def clean_name(name): @@ -292,14 +298,19 @@ def __init__(self, onnx_path, kmodel_path, dataset_path, conf_path, output_zip_f self.output_zip_file = output_zip_file def run(self): - import convertor - # 耗时操作放在这里 - shapes = get_input_shape(self.onnx_path) - if 1 == len(shapes): - shape = list(shapes.values())[0] - convertor.make(self.onnx_path, self.kmodel_path, self.dataset_path, self.conf_path,shape) - file_path = zip_with_md5(base_name=self.output_zip_file) - self.finished.emit(file_path) # 发射信号通知主线程 + try: + import convertor + # 耗时操作放在这里 + shapes = get_input_shape(self.onnx_path) + if 1 == len(shapes): + shape = list(shapes.values())[0] + convertor.make(self.onnx_path, self.kmodel_path, self.dataset_path, self.conf_path, shape) + file_path = zip_with_md5(base_name=self.output_zip_file) + self.finished.emit(file_path) # 发射信号通知主线程 + except Exception as e: + import traceback + traceback.print_exc() + print(f"!!! 子线程发生严重错误: {e} !!!", flush=True) class ModelExportApp(QWidget): @@ -705,7 +716,7 @@ def export_model(self): icon_file = self._conf["comm"]["icon_file"] if os.path.exists(icon_file): - shutil.copy(icon_file, os.path.join("model_output/icon.png")) + shutil.copy(icon_file, os.path.join("model_output", "icon.png")) # 创建空文件 open(f"model_output/app.{conf_data['conf']['application']}", "w").close() @@ -736,6 +747,12 @@ def pack(self): print("转换完成！") if __name__ == "__main__": + import traceback + # 拦截 PyQt5 的主线程异常，防止 0xC0000409 闪退 + def exception_hook(type, value, tb): + traceback.print_exception(type, value, tb) + print(f"\n[程序内部报错] {value}", flush=True) + sys.excepthook = exception_hook language_code = locale.getdefaultlocale()[0] print(f"默认语言环境: {language_code}") if language_code.startswith("zh_CN"): diff --git a/convertor.py b/convertor.py index 79a9f1b..e8d0748 100644 --- a/convertor.py +++ b/convertor.py @@ -2,35 +2,27 @@ import subprocess import argparse -import nncase -import tomlkit import sys import os import cv2 import numpy as np from pathlib import Path -swapRB = False -preprocess = False - -input_type = np.uint8 - -templs_shape = 640 - - -# setup env -result = subprocess.run(["pip", "show", "nncase"], capture_output=True) +result = subprocess.run([sys.executable, "-m", "pip", "show", "nncase"], capture_output=True) line_break = "\n" if sys.platform == "win32": line_break = "\r\n" -location_s = [i for i in result.stdout.decode().split( - line_break) if i.startswith("Location:")] -location = location_s[0].split(": ")[1] -if "PATH" in os.environ: - os.environ["PATH"] += os.pathsep + location -else: - os.environ["PATH"] = location -os.environ["NNCASE_PLUGIN_PATH"] = location +location_s = [i for i in result.stdout.decode().split(line_break) if i.startswith("Location:")] +if location_s: + location = location_s[0].split(": ")[1] + if "PATH" in os.environ: + os.environ["PATH"] += os.pathsep + location + else: + os.environ["PATH"] = location + os.environ["NNCASE_PLUGIN_PATH"] = location + +import nncase +import tomlkit class Convertor(nncase.Compiler): @@ -134,30 +126,38 @@ def gen(_dir): ''' def gen(_dir): - path = Path(_dir) - + path = Path(_dir).resolve() + count = 0 for f in path.rglob('*'): if not f.is_file(): continue - - img_path = str(f) # ؼҪ f.name + img_path = str(f) data = np.fromfile(img_path, dtype=np.uint8) templ = cv2.imdecode(data, cv2.IMREAD_COLOR) if templ is None: - print(f"[WARN] read failed: {img_path}") continue templ = process_img(templ) + count += 1 + print(f" -> 成功载入校准图 [{count}/100]: {f.name}", flush=True) yield templ def make(onnx_file, kmodel_file, dataset, toml_file, input_shape): global templs_shape templs_shape = input_shape[2] calib = [] + max_calib_images = 100 for t in gen(dataset): calib.append(t) + if len(calib) >= max_calib_images: + break + + if len(calib) == 0: + print(f"\n[错误] 严重警告：在目录 '{dataset}' 下未找到任何有效的图片！", flush=True) + print("请检查该文件夹是否存在，且里面是否确实含有 .jpg 或 .png 图片。\n", flush=True) + raise FileNotFoundError(f"未能在 {dataset} 中找到校准图片") npcalib = np.array(calib).astype(np.uint8)