Update gradio_utils/utils.py

gradio_utils/utils.py

@@ -15,7 +15,8 @@ from mmpose.core import wrap_fp16_model
 from mmpose.models import build_posenet
 from torchvision import transforms
 import matplotlib.patheffects as mpe
-
+import pickle
+from pprint import pformat as pf
 from EdgeCape import TopDownGenerateTargetFewShot
 from demo import Resize_Pad
 from EdgeCape.models import *
@@ -209,6 +210,7 @@ def process(query_img, state,
                        original_skeleton=state['skeleton'],
                        img_alpha=1.0,
                        )
+    print(pf(pickle_trick(model)))
     return out
 
 
@@ -394,3 +396,29 @@ def draw_points_on_image(image,
             )
 
     return Image.alpha_composite(image.convert("RGBA"), overlay_rgba).convert("RGB")
+
+def pickle_trick(obj, max_depth=10):
+    output = {}
+
+    if max_depth <= 0:
+        return output
+
+    try:
+        pickle.dumps(obj)
+    except (pickle.PicklingError, TypeError) as e:
+        failing_children = []
+
+        if hasattr(obj, "__dict__"):
+            for k, v in obj.__dict__.items():
+                result = pickle_trick(v, max_depth=max_depth - 1)
+                if result:
+                    failing_children.append(result)
+
+        output = {
+            "fail": obj,
+            "err": e,
+            "depth": max_depth,
+            "failing_children": failing_children
+        }
+
+    return output