소스 검색

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