mirror of https://github.com/open-mmlab/mmyolo.git
parent
f2a576d153
commit
4e8baf8a82
|
@ -6,7 +6,7 @@
|
|||
|
||||
This project is developed for easily showing assigning results. The script allows users to analyze where and how many positive samples each gt is assigned in the image.
|
||||
|
||||
Now, the script supports `YOLOv5`, `YOLOv7` and `RTMDet`.
|
||||
Now, the script supports `YOLOv5`, `YOLOv7`, `YOLOv8` and `RTMDet`.
|
||||
|
||||
## Usage
|
||||
|
||||
|
@ -26,12 +26,18 @@ YOLOv7 assigner visualization command:
|
|||
python projects/assigner_visualization/assigner_visualization.py projects/assigner_visualization/configs/yolov7_tiny_syncbn_fast_8xb16-300e_coco_assignervisualization.py -c ${checkpont}
|
||||
```
|
||||
|
||||
YOLOv8 assigner visualization command:
|
||||
|
||||
```shell
|
||||
python projects/assigner_visualization/assigner_visualization.py projects/assigner_visualization/configs/yolov8_s_syncbn_fast_8xb16-500e_coco_assignervisualization.py -c ${checkpont}
|
||||
```
|
||||
|
||||
RTMdet assigner visualization command:
|
||||
|
||||
```shell
|
||||
python projects/assigner_visualization/assigner_visualization.py projects/assigner_visualization/configs/rtmdet_s_syncbn_fast_8xb32-300e_coco_assignervisualization.py -c ${checkpont}
|
||||
```
|
||||
|
||||
${checkpont} is the checkpont file path. Dynamic label assignment is used in `YOLOv7` and `RTMDet`, model weights will affect the positive sample allocation results, so it is recommended to load the trained model weights.
|
||||
${checkpont} is the checkpont file path. Dynamic label assignment is used in `YOLOv7`, `YOLOv8` and `RTMDet`, model weights will affect the positive sample allocation results, so it is recommended to load the trained model weights.
|
||||
|
||||
If you want to know details about label assignment, you can check the [RTMDet](https://mmyolo.readthedocs.io/zh_CN/latest/algorithm_descriptions/rtmdet_description.html#id5).
|
||||
|
|
Loading…
Reference in New Issue