[GDMEC-无人机遥感研究小组]无人机遥感小组-000-数据集制备

基于labelme的无人机语义分割数据集制备

文章目录

  • 基于labelme的无人机语义分割数据集制备
    • 1. 数据获取
    • 2. 安装labelme
    • 3.利用labelme进行标注

1. 数据获取

数据集制备需要利用无人机飞行并采集标注。使用录制模式,镜头垂直向下进行拍摄,得到DJI_XXXX.MP4文件,利用如下代码,可以按照如下代码得到对应的图片(注,本代码来自另一博主,非本人原创)

import cv2
import os
#要提取视频的文件名,隐藏后缀
sourceFileName='DJI_0288'
#在这里把后缀接上
video_path = os.path.join("", "", sourceFileName+'.MP4')
times=0
#提取视频的频率,每25帧提取一个
frameFrequency=100
#输出图片到当前目录vedio文件夹下
outPutDirName=''+sourceFileName+'/'
if not os.path.exists(outPutDirName):
    #如果文件目录不存在则创建目录
    os.makedirs(outPutDirName) 
camera = cv2.VideoCapture(video_path)
while True:
    times+=1
    res, image = camera.read()
    if not res:
        print('not res , not image')
        break
    if times%frameFrequency==0:
        count = int(times/frameFrequency)
        cv2.imwrite(outPutDirName + str(count).zfill(6)+'.jpg', image)
        print(outPutDirName + str(count).zfill(6)+'.jpg')
print('图片提取结束')
camera.release()

可得到如下图像文件:
[GDMEC-无人机遥感研究小组]无人机遥感小组-000-数据集制备_第1张图片

2. 安装labelme

假定你已经安装好了anaconda,直接在prompt中新建labelme环境:conda create -n labelme python=3.8
[GDMEC-无人机遥感研究小组]无人机遥感小组-000-数据集制备_第2张图片

建好后,conda activate labelme进入labelme环境
[GDMEC-无人机遥感研究小组]无人机遥感小组-000-数据集制备_第3张图片
然后,安装labelme,pip install labelme即可

3.利用labelme进行标注

启动labelme,输入命令labelme即可
[GDMEC-无人机遥感研究小组]无人机遥感小组-000-数据集制备_第4张图片
进入界面
[GDMEC-无人机遥感研究小组]无人机遥感小组-000-数据集制备_第5张图片
具体操作流程,详见B站视频即可

你可能感兴趣的:(GDMEC-无人机遥感研究小组,无人机)