Fixing file name

This commit is contained in:
Elijah Harmon 2022-12-29 17:04:15 -05:00
parent 2aea2d4398
commit 5648375443

View File

@ -94,7 +94,7 @@ def main():
sTime = time.time()
# Loading Yolo5 Small AI Model
model = DetectMultiBackend('yolov5s.engine', device=torch.device(
model = DetectMultiBackend('yolov5s320Half.engine', device=torch.device(
'cuda'), dnn=False, data='', fp16=True)
stride, names, pt = model.stride, model.names, model.pt