|
|
@@ -129,10 +129,10 @@ I have provided a bash file `train_ddp.sh` that enables DDP training. I hope som
|
|
|
|---------------|--------------------|-------|-------|------------------------|-------------------|-------------------|--------------------|--------|
|
|
|
| YOLOv5-N | CSPDarkNet-N | 640 | 250 | 29.8 | 47.1 | 7.7 | 2.4 | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolov5_n_coco.pth) |
|
|
|
| YOLOv5-S | CSPDarkNet-S | 640 | 250 | 37.8 | 56.5 | 27.1 | 9.0 | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolov5_s_coco.pth) |
|
|
|
-| YOLOv5-M | CSPDarkNet-M | 640 | 250 | 43.5 | 62.5 | 74.3 | 25.4 | |
|
|
|
+| YOLOv5-M | CSPDarkNet-M | 640 | 250 | 43.5 | 62.5 | 74.3 | 25.4 | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolov5_m_coco.pth) |
|
|
|
| YOLOv5-L | CSPDarkNet-L | 640 | 250 | 46.7 | 65.5 | 155.6 | 54.2 | [ckpt](https://github.com/yjh0410/PyTorch_YOLO_Tutorial/releases/download/yolo_tutorial_ckpt/yolov5_l_coco.pth) |
|
|
|
|
|
|
-*For **YOLOv5-L**, increasing the batch size may improve performance. Due to my computing resources, I can only set the batch size to 16.*
|
|
|
+*For **YOLOv5-M** and **YOLOv5-L**, increasing the batch size may improve performance. Due to my computing resources, I can only set the batch size to 16.*
|
|
|
|
|
|
* YOLOX:
|
|
|
|
|
|
@@ -143,7 +143,7 @@ I have provided a bash file `train_ddp.sh` that enables DDP training. I hope som
|
|
|
| YOLOX-M | CSPDarkNet-M | 640 | 300 | | | 74.3 | 25.4 | |
|
|
|
| YOLOX-L | CSPDarkNet-L | 640 | 300 | | | 155.4 | 54.2 | |
|
|
|
|
|
|
-*For **YOLOX-L**, increasing the batch size may improve performance. Due to my computing resources, I can only set the batch size to 16.*
|
|
|
+*For **YOLOX-M** and **YOLOX-L**, increasing the batch size may improve performance. Due to my computing resources, I can only set the batch size to 16.*
|
|
|
|
|
|
* YOLOv7:
|
|
|
|
yjh0410