如何在没有互联网的情况下使用YOLOv5进行kaggle竞赛

Question

我正试图在我的python代码中使用一个名为YOLOv5的软件来进行Kaggle竞赛。唯一的问题是它要求你不要使用互联网。我已经将yolov5文件加载到kaggle代码中。但我不知道该怎么叫我的代码。

我尝试过以以下方式加载它，但始终会出现错误。

​

import torch
torch.save('yolov5x6.pt', 'yolov5')
yolov5x6_model = torch.load('yolov5')

Answer 1

Yolov5是yolo的后续版本，它是c语言中的一个神经网络库，也称为由pjreddie创建的Darknet。

它是一种目标检测模型，可以训练用于识别图像或视频中的对象。

如果您只想检测一些日常生活对象，那么您可以使用python和经过训练的权重和配置文件对图像/视频进行推断。您将在以下链接的预培训检查点部分下找到这些文件。https://github.com/ultralytics/yolov5