SeekingForJoy
/
ComDesignProject
forked from wffjwbbf/ComDesignProject
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

ignore file update

This commit is contained in:
ken4647 2022-12-03 22:35:35 +08:00
parent e8b71de1e1
commit 56fb760280
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@ -7,4 +7,4 @@
*.pth
*label*
result.txt
rank10_detection.txt
rank10_detection*.txt

6
param.py
View File

@ -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