Kaynağa Gözat

修改保存图片文件名

leon 2 ay önce
ebeveyn
işleme
6418d39c34
1 değiştirilmiş dosya ile 5 ekleme ve 4 silme
  1. 5 4
      infer.py

+ 5 - 4
infer.py

@@ -224,7 +224,7 @@ class DoorInference(object):
     
     
 if __name__ == "__main__":
-    logger.info("Start Server")
+    logger.info("====== Start Server =======")
     human_model_path = "models/work_clo_person_head_hat.pt"
     door_model_path  = "models/door_classify.pt"
     #image_path = "images/camera1_9ce1aca78db14c5807a271cb154fed5b.jpg"
@@ -246,8 +246,9 @@ if __name__ == "__main__":
             try:
                 posttime  = time.time()
                 videoTime = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())
-                filename  = videoTime + ".jpg"
-                filenameori = videoTime + "det.jpg"
+                fileTime  = time.strftime('%Y-%m-%d-%H:%M:%S',time.localtime())
+                filename  = fileTime + ".jpg"
+                filenameori = fileTime + "det.jpg"
                 logger.info(videoTime)
                 logger.info(result)
                 for res in result:
@@ -270,7 +271,7 @@ if __name__ == "__main__":
                 logger.info(result)
             except Exception as error:
                 logger.info('Error : ', str(error))
-    logger.info("EXIT")
+    logger.info("=======  EXIT  =======")
     #cv2.imwrite("result/result.jpg", image)