debug

app.py

@@ -142,7 +142,7 @@ def get_codebook_v2(quantizer):
     return codes
 
 def draw_fig_v2(ax, quantizer, data, title=""):
-    codes = get_codebook(quantizer)
+    codes = get_codebook_v2(quantizer)
     ax.scatter(data[:, 0], data[:, 1], s=60, marker="*")
     ax.scatter(codes[:, 0], codes[:, 1], s=20, c='red', alpha=0.5)
     ax.plot([-12, 12], [-12, 12], color='orange', linestyle='--', linewidth=2)
@@ -291,19 +291,6 @@ if __name__ == "__main__":
     #     description="Upload an image to see how different VQ models (BaseVQ, VQGAN, ReVQ) reconstruct it from latent codes."
     # )
 
-    # demo2 = gr.Interface(
-    #     fn=draw_reset_result,  # 不再需要 handler 包装
-    #     inputs=[
-    #         gr.Slider(label="num_data", value=16, minimum=10, maximum=20, step=1),
-    #         gr.Slider(label="num_code", value=12, minimum=8, maximum=16, step=1),
-    #     ],
-    #     outputs=[
-    #         gr.Image(label="With Reset"),
-    #         gr.Image(label="Without Reset")
-    #     ],
-    #     title="Demo 2: Codebook Reset Strategy Visualization",
-    #     description="Visualizes codebook and data movement at different training steps."
-    # )
     with gr.Blocks() as demo2:
         gr.Markdown("## Demo 2: Codebook Reset Strategy Visualization")
         gr.Markdown("Visualizes codebook and data movement at different training steps.")