leon 2 meses atrás
pai
commit
c358c32fec
2 arquivos alterados com 27 adições e 25 exclusões
  1. 1 1
      Makefile
  2. 26 24
      infer.py

+ 1 - 1
Makefile

@@ -4,7 +4,7 @@ start:
 	@ps -ef | grep infer.py | grep -v grep
 
 debug:
-	python3 infer.py
+	@python3 infer.py
 
 status:
 	@ps -ef | grep infer.py | grep -v grep

+ 26 - 24
infer.py

@@ -237,37 +237,39 @@ if __name__ == "__main__":
     ip = '172.19.152.231'
     channel = '45'
     stream = StreamCapture(ip, channel)
-    posttime = 0
+    posttime = time.time()
     for frame, ret in stream():
         if not ret:continue
         image = frame.copy()
         result = instance(image)
-        if len(result)>0 and time.time()-posttime>30:
-            videoTime = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())
-            filename  = videoTime + ".jpg"
-            filenameori = videoTime + "det.jpg"
-            logger.info(videoTime)
-            logger.info(result)
-            for res in result:
-                cv2.rectangle(image, tuple(map(int, (res.left, res.top))), tuple(map(int, (res.right, res.bottom))), (255,0, 0), 4)
-            success, encoded_image = cv2.imencode('.jpg', image)
-            content = encoded_image.tobytes()
-            successori, encoded_imageori = cv2.imencode('.jpg',frame)
-            contentori = encoded_imageori.tobytes()
-            payload = {'channel': '45',
-                        'classIndex': '8',
-                        'ip': '172.19.152.231',
-                        'videoTime': videoTime,
-                        'videoUrl': stream.stream_url}
-            files = [
-                            ('file', (filename, content, 'image/jpeg')),
-                            ('oldFile', (filenameori, contentori, 'image/jpeg')),
-                        ]
+        if len(result) > 0 and time.time() - posttime > 30:
             try:
+                posttime  = time.time()
+                videoTime = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime())
+                filename  = videoTime + ".jpg"
+                filenameori = videoTime + "det.jpg"
+                logger.info(videoTime)
+                logger.info(result)
+                for res in result:
+                    cv2.rectangle(image, tuple(map(int, (res.left, res.top))), tuple(map(int, (res.right, res.bottom))), (255,0, 0), 4)
+                success, encoded_image = cv2.imencode('.jpg', image)
+                content = encoded_image.tobytes()
+                successori, encoded_imageori = cv2.imencode('.jpg',frame)
+                contentori = encoded_imageori.tobytes()
+                payload = {'channel': '45',
+                            'classIndex': '8',
+                            'ip': '172.19.152.231',
+                            'videoTime': videoTime,
+                            'videoUrl': stream.stream_url}
+                files = [
+                                ('file', (filename, content, 'image/jpeg')),
+                                ('oldFile', (filenameori, contentori, 'image/jpeg')),
+                            ]
+            
                 result = requests.post('http://172.19.145.197/open/api/operate/upload', data=payload, files=files)
                 logger.info(result)
-            except Exception:
-                logger.info('posterror')
+            except Exception as error:
+                logger.info('Error : ', str(error))
     logger.info("EXIT")
     #cv2.imwrite("result/result.jpg", image)