今天找到一份数据集和两份源码,成功复现代码!
算法使用的是YOLO V7和YOLO V8,算法很牛。出来效果很好。
源码资料这里(网盘形式)
https://download.csdn.net/download/fzf1996/87831934
首先导入数据集
import os
from distutils.dir_util import copy_tree
fromDir = "/input/detection-dataset"
toDir = "temp"
copy_tree(fromDir,toDir)
# Create configuration
import yaml
config = {'path': 'temp/dataset',
'train': 'temp/ddataset/train',
'val': 'temp/dataset/valid',
'nc': 1,
'names': ['drone']}
with open("data.yaml", "w") as file:
yaml.dump(config, file, default_flow_style=False)
import torch
torch.cuda.empty_cache()
效果相当好!