2

9 months ago · ddb66703a1
1 changed files with 3 additions and 3 deletions
--- a/tools/cal_weight.py
+++ b/tools/cal_weight.py
@ -100,17 +100,17 @@ def down_obj_from_oss(workdir, pid, action):
				@@ -100,17 +100,17 @@ def down_obj_from_oss(workdir, pid, action):
                isMtlFile = True
                
        
-        if localfile.endswith('.obj'):
+        if "obj" in localfile:
            if isObjFile:
                print(f'目录{dirPath}已存在obj文件，跳过')
                continue
        
-        if localfile.endswith('.jpg'):  
+        if "jpg" in localfile:  
            if isJpgFile:
                print(f'目录{dirPath}已存在jpg文件，跳过')
                continue
        
-        if localfile.endswith('.mtl'):
+        if "mtl" in localfile:
            if isMtlFile:
                print(f'目录{dirPath}已存在mtl文件，跳过')
                continue