瀏覽代碼

modify YOLOvx-Small-Aug

yjh0410 2 年之前
父節點
當前提交
b5feb18053
共有 2 個文件被更改,包括 6 次插入6 次删除
  1. 4 4
      README.md
  2. 2 2
      config/data_config/transform_config.py

+ 4 - 4
README.md

@@ -158,10 +158,10 @@ python train.py --cuda -d coco --root path/to/COCO -m yolov1 -bs 16 --max_epoch
 
 | Model  | Scale | AP<sup>val<br>0.5:0.95 | AP<sup>val<br>0.5 | FLOPs<br><sup>(G) | Params<br><sup>(M) | Weight |
 |--------|-------|------------------------|-------------------|-------------------|--------------------|--------|
-| LOWDet |  320  |                        |                   |      1.05         |        1.20        |  |
-| LOWDet |  416  |                        |                   |      1.78         |        1.20        |  |
-| LOWDet |  512  |                        |                   |      2.70         |        1.20        |  |
-| LOWDet |  640  |                        |                   |      4.21         |        1.20        |  |
+| LOWDet |  320  |                        |                   |      1.05         |        1.72        |  |
+| LOWDet |  416  |                        |                   |      1.78         |        1.72        |  |
+| LOWDet |  512  |                        |                   |      2.70         |        1.72        |  |
+| LOWDet |  640  |                        |                   |      4.21         |        1.72        |  |
 
 #### Redesigned RT-DETR:
 

+ 2 - 2
config/data_config/transform_config.py

@@ -303,8 +303,8 @@ yolovx_small_trans_config = {
     'aug_type': 'yolov5',
     # Basic Augment
     'degrees': 0.0,
-    'translate': 0.2,
-    'scale': 0.9,
+    'translate': 0.1,
+    'scale': 0.5,
     'shear': 0.0,
     'perspective': 0.0,
     'hsv_h': 0.015,