Browse Source

修改

master
Zhiyong Wu 3 years ago
parent
commit
921cbc81e1
2 changed files with 3 additions and 1 deletions
  1. +2
    -1
      track.py
  2. +1
    -0
      yolov5_flask.py

+ 2
- 1
track.py View File

@@ -274,4 +274,5 @@ if __name__ == '__main__':




with torch.no_grad(): with torch.no_grad():
detect(args)
detect(args)


+ 1
- 0
yolov5_flask.py View File

@@ -15,6 +15,7 @@ from utils.datasets import *
from utils.utils import * from utils.utils import *





class Camera(BaseCamera): class Camera(BaseCamera):
video_source = 'people.mp4' video_source = 'people.mp4'




Loading…
Cancel
Save