diff --git a/tools/cal_weight.py b/tools/cal_weight.py
index 9227fec..121fbbd 100644
--- a/tools/cal_weight.py
+++ b/tools/cal_weight.py
@@ -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