关键点 旋转数据增强

目录

Deepfashion2 关键点数据

旋转后可视化

训练过程数据增强：

---

Deepfashion2 关键点数据

import glob
import os
import json
import cv2
import numpy as np
import random
from pycocotools.coco import COCO
 
class MyEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, np.integer):
            return int(obj)
        elif isinstance(obj, np.floating):
            return float(obj)
        elif isinstance(obj, np.ndarray):
            return obj.tolist()
        return super(MyEncoder, self).default(obj)
 
base_dir = r'E:\data\clothes\data_ok'
img_root = fr'{base_dir}\train'
json_root = img_root  # JSON与图像同目录
output_dir = r'D:\project\cloth\visualized&#