diff --git a/.gitignore b/.gitignore index 9a44bd5..f2d6851 100644 --- a/.gitignore +++ b/.gitignore @@ -7,4 +7,4 @@ *.pth *label* result.txt -rank10_detection.txt \ No newline at end of file +rank10_detection*.txt \ No newline at end of file diff --git a/param.py b/param.py index 2c54f94..3b34224 100644 --- a/param.py +++ b/param.py @@ -4,14 +4,14 @@ class Parameters(object): def __init__(self) -> None: self.weights_yolo = "./best.pt" self.weights_reid = "./resnet18.pth" - self.conf_thres = 0.30 + self.conf_thres = 0.40 self.iou_thres = 0.60 self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") self.nosave = False self.classes = [0] self.agnostic_nms = None self.augment = None - self.query_index = 5 - self.gallary_index= 6 + self.query_index = 1 + self.gallary_index= 2 pass \ No newline at end of file