huangxin/ailai_image_point_diff
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新了calculate_diff3.0.py,一次加载rknn，加速推理

Browse Source
This commit is contained in:
琉璃月光
2025-12-12 16:48:37 +08:00
parent 6a6f639a9a
commit 72b5052d2e
4 changed files with 227 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ailai_pc/point_error_test.py
View File

@ -6,8 +6,8 @@ from ultralytics import YOLO
# ====================== 用户配置 ======================
#MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai2/weights/best.pt'
MODEL_PATH = 'point.pt'
IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251208' # 验证集图片目录
LABEL_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251208' # 标签目录（与图片同名 .txt）
IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251212' # 验证集图片目录
LABEL_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251212' # 标签目录（与图片同名 .txt）
OUTPUT_DIR = './output_images'