# 20240703: ccpd dataset to coco format dataset.
import os
import cv2 as cv
import numpy as np
imgw = 720
imgh = 1160
imgsz = imgw, imgh
# 34
provinces = ["皖", "沪", "津", "渝", "冀", "晋", "蒙", "辽", "吉", "黑", "苏", "浙", "京", "闽", "赣", "鲁", "豫",
"鄂", "湘", "粤", "桂", "琼", "川", "贵", "云", "藏", "陕", "甘", "青", "宁", "新", "警", "学", "O"]
# 25
alphabets = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'J', 'K', 'L', 'M', 'N',
'P', 'Q', 'R', 'S', 'T', 'U', 'V','W', 'X', 'Y', 'Z', 'O']
# 35
ads = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'J', 'K', 'L', 'M', 'N', 'P', 'Q', 'R', 'S', 'T',
'U', 'V', 'W', 'X', 'Y', 'Z', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'O']
def get_plate_licenses(plate):
"""
普通蓝牌共有7位字符；新能源车牌有8位字符：https://baike.baidu.com/item/%E8%BD%A6%E7%89%8C/8347320?fr=aladdin
《新能源电动汽车牌照和普通牌照区别介绍》https://www.yoojia.com/ask/4-11906976349117851507.html
新能源汽车车牌可分为三部分：省份简称(1位汉字)十地方行政区代号(1位字母)十序号(6位)
字母“D”代表纯电动汽车；
字母“F”代表非纯电动汽车(包括插电式混合动力和燃料电池汽车等)。
:param plate:
:return:
"""
result = [provinces[int(plate[0])], alphabets[int(plate[1])]]
result += [ads[int(p)] for p in plate[2:]]
result = "".join(result)
# 新能源车牌的要求，如果不是新能源车牌可以删掉这个if
# if result[2] != 'D' and result[2] != 'F' \
# and result[-1] != 'D' and result[-1] != 'F':
# print(plate)
# print("Error label, Please check!")
# print(plate, result)
return result
def ccpd2coco(path):
dataset_path = os.path.join(path, 'CCPD2020')
green_path = os.path.join(dataset_path, 'ccpd_green')
labelpath = os.path.join(dataset_path, 'green_label')
for path, subpaths, files in os.walk(dataset_path):
# print('subpaths: ', subpaths)
# print('files: ', files)
i = 0
for filename in files:
# if i>1:
# break
# i = i + 1
print(f'file in path: {path}, subpath: {subpaths}, filename: {filename}')
annoinfo = parse_annotation(filename, labelpath)
# display(path, annoinfo)
def display(filepath, annoinfo):
filename = annoinfo['filename']
bboxes = annoinfo['bboxes'] # [xyxy]
x1, y1, x2, y2 = bboxes[0]
img = cv.imread(os.path.join(filepath, filename))
cv.rectangle(img, (x1, y1), (x2, y2), (255, 0, 0)) # (leftup, rightdown)
cv.imwrite(filename, img)
def get_bbox(size, box):
# Convert xyxy box to YOLOv5 xywh box
dw = 1. / size[0]
dh = 1. / size[1]
xc = (box[0] + box[2])*0.5*dw
yc = (box[1] + box[3])*0.5*dh
w = (box[2]-box[0])*dw
h = (box[3]-box[1])*dh
return xc, yc, w, h
def parse_annotation(filename, labelpath):
"""
:param filename:
:return: 返回标注信息info
"""
# 0014128352490421455-90_90-212&467_271&489-271&489_212&489_212&467_271&467-0_0_3_30_30_25_31_32-79-4.jpg
annotations = filename.split("-")
# print('annotations: ', annotations)
rate = annotations[0] # 车牌区域占整个画面的比例；
angle = annotations[1].split("_") # 车牌水平和垂直角度, 水平95°, 竖直113°
box = annotations[2].replace("&", "_").split("_") # 标注框左上、右下坐标，左上(154, 383), 右下(386, 473)
point = annotations[3].replace("&", "_").split("_") # 标注框四个角点坐标，顺序为右下、左下、左上、右上
plate = annotations[4].split("_") # licenses 标注框四个角点坐标，顺序为右下、左下、左上、右上
plate = get_plate_licenses(plate)
box = [int(b) for b in box] # xyxy
bbox = get_bbox(imgsz, box) # xywh
point = [int(b) for b in point]
point = np.asarray(point).reshape(-1, 2)
bboxes = [box] # [xyxy]
angles = [angle]
points = [point]
plates = [plate]
labels = ["plate"] * len(bboxes)
classid = 1 # plate
annoinfo = {"filename": filename, "bboxes": bboxes, "points": points, "labels": labels, "plates": plates, "angles": angles}
# print('rate: ', rate)
# print('angle: ', angle)
# print('box: ', box)
# print('point: ', point)
# print('plate: ', plate)
# print('bboxes: ', bboxes)
# print('labels: ', labels)
# write coco info.
info = f"{classid} {' '.join(f'{x:.6f}' for x in bbox)}\n"
labelname = os.path.join(labelpath, filename.replace('jpg', 'txt'))
labelfile = open(labelname, 'w+')
labelfile.write(info)
labelfile.close()
return annoinfo
if __name__ == "__main__":
rootpath = os.path.dirname(os.path.realpath(__file__))
ccpd2coco(rootpath)
- 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.
- 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.
- 52.
- 53.
- 54.
- 55.
- 56.
- 57.
- 58.
- 59.
- 60.
- 61.
- 62.
- 63.
- 64.
- 65.
- 66.
- 67.
- 68.
- 69.
- 70.
- 71.
- 72.
- 73.
- 74.
- 75.
- 76.
- 77.
- 78.
- 79.
- 80.
- 81.
- 82.
- 83.
- 84.
- 85.
- 86.
- 87.
- 88.
- 89.
- 90.
- 91.
- 92.
- 93.
- 94.
- 95.
- 96.
- 97.
- 98.
- 99.
- 100.
- 101.
- 102.
- 103.
- 104.
- 105.
- 106.
- 107.
- 108.
- 109.
- 110.
- 111.
- 112.
- 113.
- 114.
- 115.
- 116.
- 117.
- 118.
扫一扫
499
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
