diff --git a/tools/cal_weight.py b/tools/cal_weight.py
index 121fbbd..7206253 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 "obj" in localfile:
+        if ".obj" in localfile:
             if isObjFile:
                 print(f'目录{dirPath}已存在obj文件，跳过')
                 continue
         
-        if "jpg" in localfile:  
+        if ".jpg" in localfile:  
             if isJpgFile:
                 print(f'目录{dirPath}已存在jpg文件，跳过')
                 continue
         
-        if "mtl" in localfile:
+        if ".mtl" in localfile:
             if isMtlFile:
                 print(f'目录{dirPath}已存在mtl文件，跳过')
                 continue