remove YOLOvx-P

README.md

@@ -143,7 +143,6 @@ python train.py --cuda -d coco --root path/to/COCO -m yolov1 -bs 16 --max_epoch
 
 | Model    | Scale | Batch | AP<sup>test<br>0.5:0.95 | AP<sup>test<br>0.5 | AP<sup>val<br>0.5:0.95 | AP<sup>val<br>0.5 | FLOPs<br><sup>(G) | Params<br><sup>(M) | Weight |
 |----------|-------|-------|-------------------------|--------------------|------------------------|-------------------|-------------------|--------------------|--------|
-| YOLOvx-P |  640  | 4xb32 |                         |                    |                        |                   |                   |                    |  |
 | YOLOvx-N |  640  | 4xb32 |                         |                    |                        |                   |                   |                    |  |
 | YOLOvx-T |  640  | 4xb32 |                         |                    |                        |                   |                   |                    |  |
 | YOLOvx-S |  640  | 4xb32 |                         |                    |                        |                   |                   |                    |  |

config/model_config/yolovx_config.py

@@ -2,60 +2,6 @@
 
 
 yolovx_cfg = {
-    'yolovx_p':{
-        # ---------------- Model config ----------------
-        ## Backbone
-        'backbone': 'elannet',
-        'pretrained': True,
-        'bk_act': 'silu',
-        'bk_norm': 'BN',
-        'bk_dpw': True,
-        'width': 0.25,
-        'depth': 0.34,
-        'stride': [8, 16, 32],  # P3, P4, P5
-        'max_stride': 32,
-        ## Neck: SPP
-        'neck': 'sppf',
-        'neck_expand_ratio': 0.5,
-        'pooling_size': 5,
-        'neck_act': 'silu',
-        'neck_norm': 'BN',
-        'neck_depthwise': True,
-        ## Neck: PaFPN
-        'fpn': 'yolovx_pafpn',
-        'fpn_reduce_layer': 'conv',
-        'fpn_downsample_layer': 'conv',
-        'fpn_core_block': 'elanblock',
-        'fpn_expand_ratio': 0.5,
-        'fpn_act': 'silu',
-        'fpn_norm': 'BN',
-        'fpn_depthwise': True,
-        ## Head
-        'head': 'decoupled_head',
-        'head_act': 'silu',
-        'head_norm': 'BN',
-        'num_cls_head': 2,
-        'num_reg_head': 2,
-        'head_depthwise': True,
-        'reg_max': 16,
-        # ---------------- Train config ----------------
-        ## Input
-        'multi_scale': [0.5, 1.25],   # 320 -> 800
-        'trans_type': 'yolovx_pico',
-        # ---------------- Assignment config ----------------
-        ## Matcher
-        'matcher': {'center_sampling_radius': 2.5,
-                    'topk_candicate': 10},
-        # ---------------- Loss config ----------------
-        ## Loss weight
-        'loss_obj_weight': 1.0,
-        'loss_cls_weight': 1.0,
-        'loss_box_weight': 5.0,
-        'loss_dfl_weight': 1.0,
-        # ---------------- Train config ----------------
-        'trainer_type': 'rtmdet',
-    },
-
     'yolovx_n':{
         # ---------------- Model config ----------------
         ## Backbone