Update node_utils.py

This commit is contained in:
smthemex
2025-07-21 20:17:49 +08:00
committed by GitHub
parent 4aa54171b4
commit faf5207fb8
+2 -2
View File
@@ -33,10 +33,10 @@ def find_directories(base_path):
if not os.path.isdir(smplx_dir):
continue
# 检查两个视频文件是否存在
has_bkgd = any(f.endswith('bkgd_video.mp4') for f in os.listdir(dir_path))
# has_bkgd = any(f.endswith('bkgd_video.mp4') for f in os.listdir(dir_path))
has_smplx = any(f.endswith('smplx_video.mp4') for f in os.listdir(dir_path))
if has_bkgd and has_smplx:
if has_smplx:
directories.append(name)
return directories