无人机目标检测相关脚本
一些常用脚本文件
gorgeous(๑><๑)
一点浩然气,千里快哉风。
展开
-
【脚本转换】VisDrone数据集转COCO
import jsonimport osimport cv2import numpy as npfrom PIL import Imageimport shutilclass Vis2COCO: def __init__(self, save_path, train_ratio, category_list, is_mode="train"): self.category_list = category_list self.images = []原创 2022-04-01 16:27:28 · 897 阅读 · 0 评论 -
视频抽帧【自用代码】
import cv2import argparseimport osdef parse_args(): """ Parse input arguments """ parser = argparse.ArgumentParser(description='Process pic') parser.add_argument('--input', help='video to process', dest='input', default=None, type=原创 2022-03-12 15:28:15 · 580 阅读 · 0 评论 -
【脚本转换】VisDrone2coco
import osimport cv2from tqdm import tqdmimport json def convert_to_cocodetection(dir, output_dir): train_dir = os.path.join(dir, "VisDrone2019-DET-train") val_dir = os.path.join(dir, "VisDrone2019-DET-val") train_annotations = os.path.jo原创 2021-09-25 15:25:02 · 476 阅读 · 0 评论 -
【脚本转换】yolo2coco
"""YOLO 格式的数据集转化为 COCO 格式的数据集--root_dir 输入根路径--save_path 保存文件的名字(没有random_split时使用)--random_split 有则会随机划分数据集,然后再分别保存为3个文件。"""import osimport cv2import jsonfrom tqdm import tqdmfrom sklearn.model_selection import train_test_splitimport argparse原创 2021-09-05 18:37:22 · 724 阅读 · 0 评论 -
【脚本转换】voc2coco
#!/usr/bin/python# pip install lxmlimport sysimport osimport jsonimport xml.etree.ElementTree as ETimport globimport argparseSTART_BOUNDING_BOX_ID = 1PRE_DEFINE_CATEGORIES = None# If necessary, pre-define category and its id# PRE_DEFINE_CATE原创 2021-08-31 19:03:54 · 448 阅读 · 2 评论 -
【EDA脚本】目标检测数据分析脚本
前言在将数据送入炼丹炉之前,最重要的一步是对数据进行分析,这样的话我们可以调整相关的参数,方便我们的模型能够更好的学习。查看数据集标签数量与标签类别import jsonwith open('images-ouc-car.json','r') as f: all = json.load(f)print('标签数量: ',len(all['categories']))print('标签类别')all['categories'] 查看数据集图片数量与标签数量print('数据集图片数原创 2021-08-31 19:01:49 · 584 阅读 · 1 评论 -
【脚本转换】Python更改txt文件内容
if __name__ == "__main__": json_folder_path = '/home/wxd/darknet/data/img/' json_names = os.listdir(json_folder_path) for json_name in json_names: if json_name[-4:] == '.txt': with open(json_folder_path+json_name,'w') as原创 2021-06-01 15:04:00 · 564 阅读 · 0 评论 -
【脚本转换】VisDrone2xml
代码"""该脚本用于visdrone数据处理;将annatations文件夹中的txt标签文件转换为XML文件;txt标签内容为:<bbox_left>,<bbox_top>,<bbox_width>,<bbox_height>,<score>,<object_category>,<truncation>,<occlusion>类别:ignored regions(0), pedestrian(1原创 2021-05-27 15:13:58 · 565 阅读 · 0 评论