17 lines
523 B
Python
17 lines
523 B
Python
import torch
|
|
|
|
class Parameters(object):
|
|
def __init__(self) -> None:
|
|
self.weights_yolo = "./best.pt"
|
|
self.weights_reid = "./resnet18.pth"
|
|
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 = 1
|
|
self.gallary_index= 2
|
|
pass
|
|
|