[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+=1res, image = camera.read()if not res:print('not res , not image')breakif 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()可得到如下图像文件：
2. 安装labelme
假定你已经安装好了anaconda，直接在prompt中新建labelme环境：conda create -n labelme python=3.8
建好后，conda activate labelme进入labelme环境
然后，安装labelme，pip install labelme即可
3.利用labelme进行标注
启动labelme，输入命令labelme即可
进入界面
具体操作流程，详见B站视频即可