17 lines
667 B
Python
17 lines
667 B
Python
|
|
|||
|
|
from ultralytics import YOLO
|
|||
|
|
import os
|
|||
|
|
os.environ["GIT_PYTHON_REFRESH"] = "quiet"
|
|||
|
|
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
# # 训练数据集：
|
|||
|
|
# model = YOLO('yolov8n.pt') # 如果要训练如pose，该对应的权重即可
|
|||
|
|
# results = model.train(data='data.yaml', epochs=200)
|
|||
|
|
|
|||
|
|
#预测结果
|
|||
|
|
model = YOLO('yolov8n.pt') #常用模型yolov8n-seg.pt、yolov8n.pt、yolov8n-pose.pt
|
|||
|
|
model.predict("ultralytics/assets", save=True) #测试图片文件夹，并且设置保存True
|
|||
|
|
|
|||
|
|
#如果中断后，可以改为以下代码：
|
|||
|
|
# model = YOLO('last.pt') # last.pt文件的路径
|
|||
|
|
# results = model.train(resume=True)
|