【实例分割/目标检测】coco大Json文件转labelme小Json文件
提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档
文章目录
- 前言
- 一、Zbase64
-
- 1.Zbase64.h
- 2.Zbase64.cpp
- 二、主函数
-
- 1.coco_class.txt对应的内容如下
- 2.主函数main.cpp
- 三、效果展示
-
- 1.将从coco大json文件导出的小json文件复制到coco2017图片目录中
- 2.激活conda-labelme环境，输入labelme
- 3.进入labelme，选择opendir选择相应文件夹
- 4.既可以查看到转换完的labelme小标签
- 总结
前言
“第一次写博客，记录一下吧”
当时写这个脚本是为了方便评估模型（当时用的是tensorrt，c++），直接将solov2推理的结果保存成了labelme格式的Json文件{（所以这次的代码也是c++代码哈）也是为了方便标数据，labelme打开可以直接微调坐标成标签}，后来想通过json文件评估模型，就把coco的大json文件转换成labelme格式的json文件，方便以后用c++推理然后评估。所以废话不多说直接上代码
一、Zbase64
使用labelme需要的图片编码
1.Zbase64.h
#pragma once
#include 2.Zbase64.cpp
#include "ZBase64.h"
string ZBase64::Encode(const unsigned char* Data, int DataByte)
{
//编码表
const char EncodeTable[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
//返回值
string strEncode;
unsigned char Tmp[4] = { 0 };
int LineLength = 0;
for (int i = 0; i < (int)(DataByte / 3); i++)
{
Tmp[1] = *Data++;
Tmp[2] = *Data++;
Tmp[3] = *Data++;
strEncode += EncodeTable[Tmp[1] >> 2];
strEncode += EncodeTable[((Tmp[1] << 4) | (Tmp[2] >> 4)) & 0x3F];
strEncode += EncodeTable[((Tmp[2] << 2) | (Tmp[3] >> 6)) & 0x3F];
strEncode += EncodeTable[Tmp[3] & 0x3F];
if (LineLength += 4, LineLength == 76) { strEncode += "\r\n"; LineLength = 0; }
}
//对剩余数据进行编码
int Mod = DataByte % 3;
if (Mod == 1)
{
Tmp[1] = *Data++;
strEncode += EncodeTable[(Tmp[1] & 0xFC) >> 2];
strEncode += EncodeTable[((Tmp[1] & 0x03) << 4)];
strEncode += "==";
}
else if (Mod == 2)
{
Tmp[1] = *Data++;
Tmp[2] = *Data++;
strEncode += EncodeTable[(Tmp[1] & 0xFC) >> 2];
strEncode += EncodeTable[((Tmp[1] & 0x03) << 4) | ((Tmp[2] & 0xF0) >> 4)];
strEncode += EncodeTable[((Tmp[2] & 0x0F) << 2)];
strEncode += "=";
}
return strEncode;
}
string ZBase64::Decode(const char* Data, int DataByte, int& OutByte)
{
//解码表
const char DecodeTable[] =
{
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
62, // '+'
0, 0, 0,
63, // '/'
52, 53, 54, 55, 56, 57, 58, 59, 60, 61, // '0'-'9'
0, 0, 0, 0, 0, 0, 0,
0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12,
13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, // 'A'-'Z'
0, 0, 0, 0, 0, 0,
26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38,
39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, // 'a'-'z'
};
//返回值
string strDecode;
int nValue;
int i = 0;
while (i < DataByte)
{
if (*Data != '\r' && *Data != '\n')
{
nValue = DecodeTable[*Data++] << 18;
nValue += DecodeTable[*Data++] << 12;
strDecode += (nValue & 0x00FF0000) >> 16;
OutByte++;
if (*Data != '=')
{
nValue += DecodeTable[*Data++] << 6;
strDecode += (nValue & 0x0000FF00) >> 8;
OutByte++;
if (*Data != '=')
{
nValue += DecodeTable[*Data++];
strDecode += nValue & 0x000000FF;
OutByte++;
}
}
i += 4;
}
else// 回车换行,跳过
{
Data++;
i++;
}
}
return strDecode;
}
二、主函数
注意！！！
1.coco_class.txt对应的内容如下
0 1
1 2
2 3
3 4
4 5
5 6
6 7
7 8
8 9
9 10
10 11
11 13
12 14
13 15
14 16
15 17
16 18
17 19
18 20
19 21
20 22
21 23
22 24
23 25
24 27
25 28
26 31
27 32
28 33
29 34
30 35
31 36
32 37
33 38
34 39
35 40
36 41
37 42
38 43
39 44
40 46
41 47
42 48
43 49
44 50
45 51
46 52
47 53
48 54
49 55
50 56
51 57
52 58
53 59
54 60
55 61
56 62
57 63
58 64
59 65
60 67
61 70
62 72
63 73
64 74
65 75
66 76
67 77
68 78
69 79
70 80
71 81
72 82
73 84
74 85
75 86
76 87
77 88
78 89
79 90
2.主函数main.cpp
注意！！！
CocoJson2LabelmeJson()中需要更改的路径有
json_val2017 // 要保存的json文件目录
img_val2017 // coco图片目录
txt_path // coco标签对应字典
main()中需要更改的路径有
image_path //coco图片路径
json_path //coco大json文件路径
#include 三、效果展示
1.将从coco大json文件导出的小json文件复制到coco2017图片目录中
2.激活conda-labelme环境，输入labelme
3.进入labelme，选择opendir选择相应文件夹
4.既可以查看到转换完的labelme小标签
总结
配置release跑的会快一些，处理掩码部分可能有一些误差（可修改缩小），代码写的不简洁请谅解，有问题的话可以在下面评论，我会相应修改。