Update param.py

2022-11-25 16:25:41 +08:00 · 2022-11-25 16:25:41 +08:00 · ef6e17c88c
parent 5868fd1df9
commit ef6e17c88c
1 changed files with 1 additions and 1 deletions
--- a/param.py
+++ b/param.py
@ -3,7 +3,7 @@ import torch
 class Parameters(object):
    def __init__(self) -> None:
        self.weights_yolo = "./best.pt"
-        self.weights_reid = "./model9.pth"
+        self.weights_reid = "./resnet18.pth"
        self.conf_thres   = 0.35
        self.iou_thres    = 0.70
        self.device       = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")