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

torch

main
luanhailiang 2023-09-21 20:32:33 +08:00
parent a54d47c3c1
commit c96c2a0b6f
1 changed files with 2 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
app.py
View File

@ -11,18 +11,6 @@ from multiprocessing import Process, Event
import torch import torch
device_count = 0
if torch.cuda.is_available():
device_count = torch.cuda.device_count()
print("GPU is available.")
print("GPU device count:", torch.cuda.device_count())
print("GPU device Current:", torch.cuda.current_device())
print("GPU device name:", torch.cuda.get_device_name(torch.cuda.current_device()))
else:
print("GPU is not available.")
app = Flask(__name__) app = Flask(__name__)
workers = {} workers = {}
@ -73,7 +61,8 @@ def worker(id,source,region=None,stream=False):
model = YOLO('./best.pt') model = YOLO('./best.pt')
balls = {} balls = {}
count = 0 count = 0
print(source) device_count = torch.cuda.device_count()
print(device_count,source)
try: try:
cap = cv2.VideoCapture(source) cap = cv2.VideoCapture(source)
if not cap.isOpened(): if not cap.isOpened():