From ef6e17c88caf35ef4cc5dffeb81918b9dce66a04 Mon Sep 17 00:00:00 2001 From: wffjwbbf <3502185053@qq.com> Date: Fri, 25 Nov 2022 16:25:41 +0800 Subject: [PATCH] Update param.py --- param.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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")