Update gradio_utils/utils.py

gradio_utils/utils.py

@@ -1,3 +1,86 @@
+BEST
+markov_bias_dinov2_with_image_attn_adj_supervise.py: 91.41
+markov_bias_dinov2_with_image_attn_adj_no_entrop.py: 90.28
+markov_bias_dinov2_with_image_attn_adj.py: 89.2
+markov_bias_dinov2_no_image_attn_adj_small_entrop.py: 90.07
+markov_bias_dinov2_no_image_attn_adj_no_entrop.py: 89.53
+markov_bias_dinov2_no_image_attn_adj_entrop_supervise.py: 89.83
+markov_bias_dinov2_with_image_attn_adj_small_entrop.py: 90.12
+markov_bias_dinov2_with_image_attn_adj_entrop_supervise.py: 89.66
+markov_bias_dinov2_no_image_attn_adj_supervise.py: 90.87
+markov_bias_dinov2_no_image_attn_adj.py: 88.81
+
+LATEST
+markov_bias_dinov2_with_image_attn_adj_supervise.py: 91.28
+markov_bias_dinov2_with_image_attn_adj_no_entrop.py: 91.05
+markov_bias_dinov2_with_image_attn_adj.py: 81.32
+markov_bias_dinov2_no_image_attn_adj_small_entrop.py: 90.07
+markov_bias_dinov2_no_image_attn_adj_no_entrop.py: 90.99
+markov_bias_dinov2_no_image_attn_adj_entrop_supervise.py: 89.83
+markov_bias_dinov2_with_image_attn_adj_small_entrop.py: 90.89
+markov_bias_dinov2_with_image_attn_adj_entrop_supervise.py: 89.66
+markov_bias_dinov2_no_image_attn_adj_supervise.py: 90.99
+markov_bias_dinov2_no_image_attn_adj.py: 89.1
+
+
+
+
+self.detach().cpu().numpy() if self.dtype.is_numpy_compatible else self
+
+ sed 's/nameserver 10.*/nameserver 132.66.150.2/' /etc/resolv.conf >/tmp/R$$ ;  cp /tmp/R$$ /etc/resolv.conf
+ 
+ 
+
+Scores
+markov_bias_mlp_gt.py: 92.17
+markov_bias_mlp_supervise_encoder.py: 91.21
+markov_bias_mlp_small_entrop.py: 88.66
+markov_bias_mlp_small_entrop_support_pe.py: 87.57
+markov_bias_mlp_no_entrop.py: 90.66
+markov_bias_mlp_gt_no_heatmap.py: 92.46
+markov_bias_mlp_gt_no_encoder.py: 89.22
+markov_bias_mlp_small_entrop_support_pe_encoder.py: 91.36
+markov_bias_mlp_small_entrop_encoder.py: 89.72
+markov_bias_mlp_gt_no_encoder_deform_attn.py: 91.42
+markov_bias_mlp_supervise.py: 91.45
+
+
+markov_bias_mlp_gt.py: 92.39
+markov_bias_mlp_supervise_encoder.py: 91.25
+markov_bias_mlp_small_entrop.py: 88.66
+markov_bias_mlp_small_entrop_support_pe.py: 87.57
+markov_bias_mlp_no_entrop.py: 90.82
+markov_bias_mlp_gt_no_heatmap.py: 92.44
+markov_bias_mlp_gt_no_encoder.py: 91.29
+markov_bias_mlp_small_entrop_support_pe_encoder.py: 91.02
+markov_bias_mlp_small_entrop_encoder.py: 91.04
+markov_bias_mlp_gt_no_encoder_deform_attn.py: 91.42
+markov_bias_mlp_supervise.py: 91.13
+
+
+
+Scores
+
+markov_bias_attn_no_heatmap_supervise.py: 91.49
+markov_bias_attn_no_heatmap_small_entrop.py: 91.01
+
+one_stage_small_gt.py: 89.62
+one_stage_small_mlp.py: 89.59
+one_stage_small_attn.py: 89.17
+
+one_stage_attn_skeleton_encoder_only.py: 91.67
+one_stage_gt.py: 92.49
+one_stage_mlp.py: 91.73
+one_stage_no_attn_bias_skeleton_attn.py: 91.75
+one_stage_mlp_skeleton_encoder_only.py: 91.64
+one_stage_no_graphs_gt.py: 91.09
+one_stage_attn.py: 91.65
+one_stage_no_attention_bias_gt.py: 92.65
+
+
+TO DO:
+Eval: markov_additional, auxilery_decoder_feat_supervis, edge_features
+
 import copy
 import spaces
 import json
@@ -22,7 +105,6 @@ from EdgeCape.models import *
 
 
 def process_img(support_image, global_state):
-    print("process_img", global_state)
     global_state['images']['image_orig'] = support_image
     if global_state["load_example"]:
         global_state["load_example"] = False
@@ -107,7 +189,7 @@ def plot_results(support_img, query_img, support_kp, support_w, query_kp, query_
         plt.axis('off')  # command for hiding the axis.
         return plt
 
-@spaces.GPU(duration=30)
+@spaces.GPU
 def process(query_img, state,
             cfg_path='configs/test/1shot_split1.py',
             checkpoint_path='ckpt/1shot_split1.pth'):
@@ -235,7 +317,7 @@ def update_examples(support_img, query_image, global_state_str):
                                          )
     skel_image = image_draw.copy()
     example_state['images']['image_skel'] = skel_image
-    print("update_examples", example_state)
+    
     return (support_img,
             kp_image,
             skel_image,
@@ -267,7 +349,6 @@ def get_closest_point_idx(pts_list, xy):
 
 
 def reset_skeleton(global_state):
-    print("reset_skeleton")
     image = global_state["images"]["image_kp"]
     global_state["images"]["image_skel"] = image
     global_state["skeleton"] = []
@@ -277,7 +358,6 @@ def reset_skeleton(global_state):
 
 
 def reset_kp(global_state):
-    print("reset_kp")
     image = global_state["images"]["image_orig"]
     global_state["images"]["image_kp"] = image
     global_state["images"]["image_skel"] = image