19 lines
574 B
Python
19 lines
574 B
Python
import torch
|
|
|
|
class Parameters(object):
|
|
def __init__(self) -> None:
|
|
self.weights = "./best.pt"
|
|
self.img_size = 640
|
|
self.conf_thres = 0.25
|
|
self.iou_thres = 0.45
|
|
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.update = None
|
|
self.project = 'runs/detect'
|
|
self.name = 'exp'
|
|
self.exist_ok = None
|
|
pass
|
|
|