添加状态分类和液面分割
This commit is contained in:
@ -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,
|
||||
)
|
||||
Reference in New Issue
Block a user