gaohang
/
jjsos_JJdetection
1
0
Fork 0
Code Issues Pull Requests Packages Projects 1 Releases Wiki Activity

Default Changelist

main
tongminqi 2026-02-28 16:05:27 +08:00
parent 36e5a67e16
commit f4cc087b15
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train_segmentation.py
View File

@ -84,7 +84,7 @@ def main():
# - 类别: 6个（搭电线、拖车上有车、仪表盘、轮胎、平安马甲、检测仪） # - 类别: 6个（搭电线、拖车上有车、仪表盘、轮胎、平安马甲、检测仪）
train_segmentation_model( train_segmentation_model(
dataset_path="datasets/xianshiping/data.yaml", dataset_path="datasets/xianshiping/data.yaml", # 数据集配置文件路径
model_name="yolov8n-seg.pt", # 使用YOLOv8 nano分割模型 model_name="yolov8n-seg.pt", # 使用YOLOv8 nano分割模型
epochs=150, # 1223张照片，建议150轮（配合早停） epochs=150, # 1223张照片，建议150轮（配合早停）
batch=8, # CPU/MPS建议8，GPU可以16-32 batch=8, # CPU/MPS建议8，GPU可以16-32