复现成功！深度学习无人机目标检测！_复现无人机目标检测算法

今天找到一份数据集和两份源码，成功复现代码！
算法使用的是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)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

import torch
torch.cuda.empty_cache()
1
2

效果相当好！