🐞 fix: misc

animation/a-pose.fbx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ff64cd4eb4bbef96951b9ca7f51400a9fa38d9bbd85bbea9955a87e4793a3c93
+size 392780

animation/da-pose.fbx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c052c2a47927f9b83c99b37a216acce793e14d81c34fbe601e6b51a331139545
+size 392844

apply_animation.py

@@ -3,36 +3,23 @@ from glob import glob
 
 from tqdm import tqdm
 
-from utils import HiddenPrints, bpy, load_mixamo_anim, remove_all, reset, select_objs, update
-
 if __name__ == "__main__":
     character_dir = "./character_vroid_refined"
     animation_dir = "./animation"
     output_dir = "./animated_vroid"
     os.makedirs(output_dir, exist_ok=True)
 
-    reset()
     character_list = sorted(glob(os.path.join(character_dir, "*.fbx")))
     assert character_list
     animation_list = sorted(glob(os.path.join(animation_dir, "*.fbx")))
     assert animation_list
+    get_base_name = lambda s: os.path.splitext(os.path.basename(s))[0]
     for char_file in tqdm(character_list, dynamic_ncols=True, desc="Character"):
         for anim_file in tqdm(animation_list, dynamic_ncols=True, leave=False, desc="Animation"):
-            with HiddenPrints():
-                remove_all()
-                objs = load_mixamo_anim(char_file, anim_file, do_retarget=True, inplace=False)
-                update()
-                get_base_name = lambda s: os.path.splitext(os.path.basename(s))[0]
-                output_filename = f"{get_base_name(char_file)}-{get_base_name(anim_file)}"
-                # bpy.ops.wm.save_as_mainfile(filepath=os.path.join(output_dir, f"{output_filename}.blend"))
-                select_objs(objs, deselect_first=True)
-                bpy.ops.export_scene.fbx(
-                    filepath=os.path.join(output_dir, f"{output_filename}.fbx"),
-                    check_existing=False,
-                    use_selection=True,
-                    use_triangles=True,
-                    add_leaf_bones=False,
-                    bake_anim=True,
-                    # path_mode="COPY",
-                    # embed_textures=True,
-                )
+            output_filename = f"{get_base_name(char_file)}-{get_base_name(anim_file)}"
+            if os.path.isfile(os.path.join(output_dir, f"{output_filename}.fbx")):
+                continue
+            # Call subprocess to avoid accumulation bugs that gradually slow down the process
+            os.system(
+                f"python apply_animation_blender.py --char_path {char_file} --anim_path {anim_file} --output_dir {output_dir}"
+            )

apply_animation_blender.py

@@ -0,0 +1,31 @@
+import argparse
+import os
+
+from utils import HiddenPrints, bpy, load_mixamo_anim, remove_all, reset, select_objs, update
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--char_path", type=str, required=True)
+    parser.add_argument("--anim_path", type=str, required=True)
+    parser.add_argument("--output_dir", type=str, required=True)
+    args = parser.parse_args()
+
+    get_base_name = lambda s: os.path.splitext(os.path.basename(s))[0]
+    output_filename = f"{get_base_name(args.char_path)}-{get_base_name(args.anim_path)}"
+    with HiddenPrints():
+        # remove_all()
+        reset()
+        objs = load_mixamo_anim(args.char_path, args.anim_path, do_retarget=True, inplace=False)
+        update()
+        # bpy.ops.wm.save_as_mainfile(filepath=os.path.join(args.output_dir, f"{output_filename}.blend"))
+        select_objs(objs, deselect_first=True)
+        bpy.ops.export_scene.fbx(
+            filepath=os.path.join(args.output_dir, f"{output_filename}.fbx"),
+            check_existing=False,
+            use_selection=True,
+            use_triangles=True,
+            add_leaf_bones=False,
+            bake_anim=True,
+            # path_mode="COPY",
+            # embed_textures=True,
+        )

bones_vroid.fbx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:39907f8772741bcc892c581a38e36bf6d46968997a19e16abc13b8317a9e51da
+size 65756

character_refine.py

@@ -15,7 +15,7 @@ from utils import (
 )
 
 
-def rename_maxiamo_bone(armature_obj):
+def rename_mixamo_bone(armature_obj):
     """Replace name like 'mixamorig10:xxx' to 'mixamorig:xxx, so that character can be correctly animated.'"""
     import re
 
@@ -59,7 +59,7 @@ if __name__ == "__main__":
                 assert len(armature_obj) == 1, "Armature number is not 1"
                 armature_obj = armature_obj[0]
                 if unposed_list is None or file in unposed_list:
-                    rename_maxiamo_bone(armature_obj)
+                    rename_mixamo_bone(armature_obj)
                 rest_bones, bones_idx_dict = get_rest_bones(armature_obj)
 
                 # mesh_quads2tris(get_all_mesh_obj(obj_list))

utils.py

@@ -267,8 +267,8 @@ def enable_arp(armature_obj: Object, addon_path=os.path.join(os.path.dirname(__f
 
     assert os.path.isfile(os.path.join(addon_path, "__init__.py")), "Auto-Rig Pro not found"
     dirname, addon_name = os.path.split(addon_path)
-    if addon_name in get_enabled_addons():
-        return
+    # if addon_name in get_enabled_addons():
+    #     return
     sys.path.insert(0, dirname)
     with Mode("POSE", armature_obj):
         # import addon_utils