class YOLO(object) 初始化函数感觉少了条语句 about yolov4-pytorch

[dsp6414]

而且代码仅用self.cuda判断是有bug的，
比如self.cuda为True, 但机器如果没有gpu或torch是cpu版本的，执行过程就会有bug

[bubbliiiing]

1、代码的哪个地方
2、什么显示不出来
3、self.cuda只是为了判断是否用cuda，自由度更高，我在训练东西的时候我都可以把Cuda设置成False，这个时候我就能预测或者训练试试看代码是否正确。如果机器没有GPU却不知道设置CUDA的话，我觉得可能基础还要加强一下。

[dsp6414]

根目录yolo.py里Yolo类的代码。

[bubbliiiing]

class YOLO(object):
_defaults = {
"model_path": 'model_data/yolo4_weights.pth',
"anchors_path": 'model_data/yolo_anchors.txt',
"classes_path": 'model_data/coco_classes.txt',
"model_image_size" : (416, 416, 3),
"confidence": 0.5,
"cuda": True
}

@classmethod
def get_defaults(cls, n):
    if n in cls._defaults:
        return cls._defaults[n]
    else:
        return "Unrecognized attribute name '" + n + "'"

#---------------------------------------------------#
#   初始化YOLO
#---------------------------------------------------#
def __init__(self, **kwargs):
    self.__dict__.update(self._defaults)
    self.class_names = self._get_class()
    self.anchors = self._get_anchors()
    self.generate()

[bubbliiiing]

不是有吗

[dsp6414]

我明白你的意思，都在default里设置参数，像我第一问那样传参创建就没法用了

[dsp6414]

只是建议哈

[bubbliiiing]

没听太懂…不是在init的时候已经update过了吗

[dsp6414]

我意思是你固定死了参数，创建时传参怎么办，每次来yolo.py文件里改吗？算没没说哈😄

[cmdbug]

加上这个：self.dict.update(kwargs)
然后在实例的时候可以直接传参数就不用改代码了
yolo = YOLO({'model_path': 'model_data/yolo4_weights.pth',
'classes_path': 'model_data/coco_classes.txt'})

[cmdbug]

加上这个：self.dict.update(kwargs) 然后在实例的时候可以直接传参数就不用改代码了 yolo = YOLO({'model_path': 'model_data/yolo4_weights.pth',
'classes_path': 'model_data/coco_classes.txt'})

哎，，怎么字典前面的2个星号不显示。。