PyTorch：The “freeze_support()” line can be omitted

最新推荐文章于 2024-03-08 11:02:00 发布

湫兮之风

最新推荐文章于 2024-03-08 11:02:00 发布

阅读量442

点赞数 8

分类专栏：深度学习文章标签：深度学习机器学习 python

本文链接：https://blog.csdn.net/Ethan_Rich/article/details/135159806

版权

深度学习专栏收录该内容

12 篇文章 0 订阅

订阅专栏

出现这个问题将所运行函数放到main函数中运行，不然会一直创建线程。

from ultralytics import YOLO
import os
os.environ['KMP_DUPLICATE_LIB_OK']='True'



if __name__ == '__main__':
    # Load a model
    model = YOLO("yolov8s.yaml")  # build a new model from scratch
    model = YOLO("yolov8s.pt")  # load a pretrained model (recommended for training)

    # Use the model
    model.train(data="data.yaml", epochs=100, batch=16, imgsz=640, workers=8, device=0,
                task='detect')  # train the model
    # metrics = model.val()  # evaluate model performance on the validation set
    # results = model("https://ultralytics.com/images/bus.jpg")  # predict on an image
    # path = model.export(format="onnx")  # export the model to ONNX formaty