添加状态分类和液面分割

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

View File

@ -7,16 +7,14 @@ if __name__ == '__main__':
results = model.train(
data='point_data.yaml',
imgsz=1280,
epochs=100,
epochs=300,
batch=4,
workers=10,
device='0',
project='runs/train',
project='runs/train/point',
name='exp_pose',
exist_ok=False,
optimizer='AdamW',
lr0=0.001,
patience=20,
# ✅ 关键:指定关键点数量
#kpt_shape=[4, 3], # [数量, 坐标维度],您的任务是 4 个关键点 (x,y)
patience=100,
)