更新最新直接用point_diff_main
This commit is contained in:
@ -5,10 +5,10 @@ from ultralytics import YOLO
|
|||||||
|
|
||||||
# ====================== 用户配置 ======================
|
# ====================== 用户配置 ======================
|
||||||
#MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai2/weights/best.pt'
|
#MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai2/weights/best.pt'
|
||||||
MODEL_PATH = 'best.pt'
|
MODEL_PATH = 'pointn.pt'
|
||||||
IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226' # 验证集图片目录
|
IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/test' # 验证集图片目录
|
||||||
LABEL_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226' # 标签目录(与图片同名 .txt)
|
LABEL_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/test' # 标签目录(与图片同名 .txt)
|
||||||
OUTPUT_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226/output_images'
|
OUTPUT_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/test/output_images'
|
||||||
|
|
||||||
|
|
||||||
IMG_EXTENSIONS = {'.png', '.jpg', '.jpeg', '.bmp', '.tiff', '.tif', '.webp'}
|
IMG_EXTENSIONS = {'.png', '.jpg', '.jpeg', '.bmp', '.tiff', '.tif', '.webp'}
|
||||||
|
|||||||
@ -37,7 +37,7 @@ def init_rknn_once(model_path):
|
|||||||
print(f"[ERROR] Failed to load RKNN: {ret}")
|
print(f"[ERROR] Failed to load RKNN: {ret}")
|
||||||
_rknn_instance = None
|
_rknn_instance = None
|
||||||
return None
|
return None
|
||||||
ret = _rknn_instance.init_runtime(core_mask=RKNNLite.NPU_CORE_0)
|
ret = _rknn_instance.init_runtime()
|
||||||
if ret != 0:
|
if ret != 0:
|
||||||
print(f"[ERROR] Failed to init RKNN runtime: {ret}")
|
print(f"[ERROR] Failed to init RKNN runtime: {ret}")
|
||||||
_rknn_instance = None
|
_rknn_instance = None
|
||||||
|
|||||||
Reference in New Issue
Block a user