diff --git a/param.py b/param.py index aeadb73..ff8f530 100644 --- 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")