Update pred_color.py

pred_color.py

@@ -98,6 +98,7 @@ def single_pred_features(image):
         image = Image.fromarray(image).convert("RGB")
     else:
         image = image.convert("RGB")
+    image = image.resize((512, 512))
     cv2_image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
     face_detector.set_input_shape(image.size[1], image.size[0])
     features = face_detector.inference(image)