diff --git a/yolo.py b/yolo.py index 425a505..265af0a 100644 --- a/yolo.py +++ b/yolo.py @@ -69,6 +69,7 @@ def __init__(self, **kwargs): self.__dict__.update(self._defaults) for name, value in kwargs.items(): setattr(self, name, value) + self._defaults[name] = value #---------------------------------------------------# # 获得种类和先验框的数量