ソースを参照

debug yolovx_n on coco

yjh0410 2 年 前
コミット
7a4e5959a4
1 ファイル変更0 行追加1 行削除
  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