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

添加状态分类和液面分割

Browse Source
This commit is contained in:
琉璃月光
2025-09-01 14:14:18 +08:00
parent 6e553f6a20
commit ad52ab9125
2379 changed files with 102501 additions and 1465 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
angle_base_seg/test_seg_angle_f1.py
View File

@ -4,8 +4,8 @@ import numpy as np
import os
# ------------------ 配置 ------------------
model_path = '../ultralytics_yolo11-main/runs/train/exp4/weights/best.pt'
img_folder = '/home/hx/yolo/ultralytics_yolo11-main/dataset1/test'
model_path = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/seg/exp3/weights/best.pt'
img_folder = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/seg/dataset2/test'
output_mask_dir = 'output_masks1'
os.makedirs(output_mask_dir, exist_ok=True)
@ -139,9 +139,9 @@ def process_image(img_path, output_dir):
composite_mask = np.maximum(composite_mask, obj_mask)
# 保存合并掩码
mask_save_path = os.path.join(output_dir, f'mask_{name_only}.png')
cv2.imwrite(mask_save_path, composite_mask)
print(f"✅ 掩码已保存: {mask_save_path}")
#mask_save_path = os.path.join(output_dir, f'mask_{name_only}.png')
#cv2.imwrite(mask_save_path, composite_mask)
#print(f"✅ 掩码已保存: {mask_save_path}")
if len(rotated_rects) < 2:
print(f"⚠️ 检测到的对象少于2个（共{len(rotated_rects)}个）: {filename}")