design design batch

app/service/design_batch/pipeline/loading.py

@@ -1,6 +1,9 @@
+import io
 import logging
 
 import cv2
+import numpy as np
+from PIL import Image
 
 from app.service.utils.new_oss_client import oss_get_image
 
@@ -71,6 +74,8 @@ class LoadImage:
             keypoint = 'head_point'
         elif name == 'earring':
             keypoint = 'ear_point'
+        elif name == 'accessories':
+            keypoint = "accessories"
         else:
             raise KeyError(f"{name} does not belong to item category list: blouse, outwear, dress, trousers, skirt, "
                            f"bag, shoes, hairstyle, earring.")