今天找到一份数据集和两份源码,成功复现代码!
算法使用的是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()

效果相当好!

更多推荐