mirrors
/
mmyolo
mirror of https://github.com/open-mmlab/mmyolo.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix typo for assigner visualization (#509)

pull/511/head
yechenzhi 2023-02-04 13:46:14 +08:00 committed by GitHub
parent 8237f19bf8
commit 6acde82ec8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
projects/assigner_visualization/README.md
View File

@ -13,5 +13,5 @@ Now, the script only support `YOLOv5` .
### Command
```shell
python projects/assigner_visualization/assigner_visualization.py projects/assigner_visualization/configs/yolov5_s-v61_syncbn_fast_8xb16-300e_coco_assignervisualization.py `
python projects/assigner_visualization/assigner_visualization.py projects/assigner_visualization/configs/yolov5_s-v61_syncbn_fast_8xb16-300e_coco_assignervisualization.py
```

2
projects/assigner_visualization/visualization/assigner_visualizer.py
View File

@ -175,7 +175,7 @@ class YOLOAssignerVisualizer(DetLocalVisualizer):
# The PALETTE in the dataset_meta is required
assert self.dataset_meta is not None
palette = self.dataset_meta['PALETTE']
palette = self.dataset_meta['palette']
x = ((grid_x_inds + offset) * stride).long()
y = ((grid_y_inds + offset) * stride).long()
center = torch.stack((x, y), dim=-1)