main
luanhailiang 2023-09-22 15:25:34 +08:00
parent 089b4dc651
commit 2d40d4251c
1 changed files with 2 additions and 1 deletions

3
app.py
View File

@ -61,10 +61,11 @@ def worker(id,source,region=None,stream=False):
region = 10
else:
region = int(region)
print("start loading model...",id,source,region)
model = YOLO('./best.pt')
print("start loaded model!!!")
balls = {}
count = 0
print(source)
try:
cap = cv2.VideoCapture(source)
if not cap.isOpened():