Clean code

.gitignore

@@ -1,3 +1,5 @@
 *.pyc
 demo_out
-logs/
+logs/
+.vscode
+checkpoints

app.py

@@ -109,39 +109,6 @@ def inference(img: Dict)-> Tuple[Union[np.ndarray|None], List[str]]:
             return (None, [])
 
 
-# with gr.Blocks(title="AniMer", css=".gradio-container") as demo:
-
-#     gr.HTML("""<div style="font-weight:bold; text-align:center; color:royalblue;">AniMer</div>""")
-
-#     with gr.Row():
-#         with gr.Column():
-#             input_image = gr.ImageEditor(label="Input image", sources=["upload", "clipboard"],
-#                                          brush=False, eraser=False, layers=False, transforms="crop",
-#                                          interactive=True,
-#                                          )
-#             crop_image = gr.Image(label="Crop image", sources=[])
-#             input_image.change(predict, outputs=crop_image, inputs=input_image, show_progress="hidden")
-#         with gr.Column():
-#             output_image = gr.Image(label="Overlap image")
-#             output_mesh = gr.Model3D(display_mode="wireframe", label="3D Mesh")
-
-#     gr.HTML("""<br/>""")
-
-#     with gr.Row():
-#         send_btn = gr.Button("Inference")
-#         send_btn.click(fn=inference, inputs=[crop_image], outputs=[output_image, output_mesh])
-
-#     example_images = gr.Examples([
-#         ['example_data/000000015956_horse.png'], 
-#         ['example_data/n02101388_1188.png'],
-#         ['example_data/n02412080_12159.png'],
-#         ['example_data/000000101684_zebra.png']
-#         ], 
-#         inputs=[input_image])
-
-# demo.launch(debug=True)
-
-
 demo = gr.Interface(
     fn=inference,
     analytics_enabled=False,
@@ -152,10 +119,9 @@ demo = gr.Interface(
         gr.Image(label="Overlap image"),
         gr.Model3D(display_mode="wireframe", label="3D Mesh"),
     ],
-    title="AniMer",
+    title="AniMer: Animal Pose and Shape Estimation Using Family Aware Transformer",
     description="""
-    # AniMer: Animal Pose and Shape Estimation Using Family Aware Transformer
-    https://luoxue-star.github.io/AniMer_project_page/
+    Project page: https://luoxue-star.github.io/AniMer_project_page/
     ## Steps for Use
     1. **Input**: Select an example image or upload your own image.
     2. **Crop**: Crop the animal in the image (Otherwise, the result may be poor.)

config/config.yaml

@@ -38,6 +38,7 @@ EXTRA:
   NUM_LOG_SAMPLES_PER_IMAGE: 4
   PELVIS_IND: 0
 MODEL:
+  USE_CLS: true
   IMAGE_SIZE: 256
   IMAGE_MEAN:
   - 0.485