luanhailiang
/
billiard
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

4

main
luanhailiang 2023-09-23 00:49:42 +08:00
parent 1dab8ae3e1
commit c496ffc1ba
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
app.py
View File

@ -14,7 +14,6 @@ from multiprocessing import Process, Event
app = Flask(__name__) app = Flask(__name__)
device_count = torch.cuda.device_count()
workers = {} workers = {}
@app.route('/start/<id>') @app.route('/start/<id>')
@ -63,7 +62,6 @@ def worker(id,source,region=None,stream=False):
region = 10 region = 10
else: else:
region = int(region) region = int(region)
print("GPU device count:", device_count)
print("start loading model...",id,source,region) print("start loading model...",id,source,region)
model = YOLO('./best.pt') model = YOLO('./best.pt')
print("start loaded model!!!") print("start loaded model!!!")
@ -95,8 +93,7 @@ def worker(id,source,region=None,stream=False):
print("Error opening video stream....") print("Error opening video stream....")
break break
continue continue
print("device_count",int(id)%device_count) result = model.track(frame,show=False,stream=False,persist=True,device=int(id)%4)
result = model.track(frame,show=False,stream=False,persist=True,device=int(id)%device_count)
use += " track:"+str((time.time()*1000) - bgn) use += " track:"+str((time.time()*1000) - bgn)
del(ret) del(ret)
del(frame) del(frame)