Explorar el Código

debug yolovx_n on coco

yjh0410 hace 2 años
padre
commit
7a4e5959a4
Se han modificado 1 ficheros con 0 adiciones y 1 borrados
  1. 0 1
      models/detectors/yolovx/yolovx.py

+ 0 - 1
models/detectors/yolovx/yolovx.py

@@ -29,7 +29,6 @@ class YOLOvx(nn.Module):
         self.cfg = cfg
         self.device = device
         self.stride = cfg['stride']
-        self.reg_max = cfg['reg_max']
         self.num_classes = num_classes
         self.trainable = trainable
         self.conf_thresh = conf_thresh