PaddleOCR的方向分类器cls如何造数据
==================================================================
1.把照片数据按照奇数偶数分成两个类别,并分别保存到对应的路径并且写入到标签文件中
import os
import cv2
class BBox(object): # 图象尺寸的框,上下左右
def __init__(self, bbox):
self.left = bbox[0]
self.top = bbox[1]
self.right = bbox[2]
self.bottom = bbox[3]
def rotate_imgs(source_path, label_txt):
photos = os.listdir(source_path) # 列举路径下的所有照片名称
print(photos) # 打印所有照片名称的列表
for i, v in enumerate(photos):
# print(i, v) # 图片索引和名称
img_path = source_path + v # 图片路径
img = cv2.imread(img_path) # 从每张图片路径中读到像素
# print(img) # 查看是否可以显示像素值
h, w, c = img.shape # 图片尺寸
box = [0, 0, w, h] # 图象大小的框
print(box) # 打印图象框
bbox = BBox(box)