forked from wffjwbbf/ComDesignProject
14 lines
404 B
Python
14 lines
404 B
Python
import torch
|
|
|
|
class Parameters(object):
|
|
def __init__(self) -> None:
|
|
self.weights = "./best.pt"
|
|
self.conf_thres = 0.35
|
|
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
|
|
pass
|
|
|