角蜂鸟 Console下可运行的Demo

电子月

角蜂鸟Demo要在图形界面下才能跑，有些场景做演示不合适，且浪费资源。最好能在命令行下跑。我用pygame，实现在命令行下运行，就可以看到视频的Demo。

1. import numpy as np,sys,cv2
   
2. sys.path.append('../../api/')
   
3. import hsapi as hs
   
4. 
   
5. WEBCAM = False # Set to True if use Webcam
   
6. net = hs.HS('ObjectDetector', zoom = True, verbose = 2)
   
7. if WEBCAM: video_capture = cv2.VideoCapture(0)
   
8. 
   
9. import time, pygame
   
10. from pygame.locals import *
    
11. # 显示器分辨率，我用的是竖屏，所以是w=1080 h=1920
    
12. screen_size=(1080,1920)
    
13. 
    
14. def init():
    
15.         pygame.init()
    
16.         screen = pygame.display.set_mode(screen_size)
    
17.         return screen
    
18. 
    
19. def update(screen,surface):
    
20.         screen.blit(surface,((screen_size[0]-surface.get_width())/2, (screen_size[1]-surface.get_height())/2))
    
21.         pygame.display.update()
    
22. 
    
23. def eventCheck():
    
24.         for event in pygame.event.get():
    
25.                 if event.type in [QUIT,KEYDOWN]:
    
26.                         pygame.quit()
    
27.                         sys.exit()
    
28. screen=init()
    
29. 
    
30. while True:
    
31.         if WEBCAM: _, img = video_capture.read()
    
32.         else: img = None
    
33.         result = net.run(img)
    
34.         img = net.plotSSD(result)
    
35.         img = cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
    
36.         s = img.tostring()
    
37.         p = pygame.image.frombuffer(s,(img.shape[1],img.shape[0]),"RGB")
    
38.         eventCheck()
    
39.         update(screen,p)

复制代码

作者：_49_
來源：简书