[Fix] Fix import error in `optimize_anchors.py` (#452)

* [Fix] Fix bugs in the output annotation file when  is set.

* Update labelme2coco.py

* Update labelme2coco.py

* Improve code

* [Fix]Fix import error in optimize_anchors.py

* [Fix]Fix import error in optimize_anchors.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

tools/analysis_tools/optimize_anchors.py

@@ -35,7 +35,6 @@ from typing import Tuple
 
 import numpy as np
 import torch
-from mmdet.datasets import build_dataset
 from mmdet.structures.bbox import (bbox_cxcywh_to_xyxy, bbox_overlaps,
                                    bbox_xyxy_to_cxcywh)
 from mmdet.utils import replace_cfg_vals, update_data_root
@@ -46,6 +45,7 @@ from mmengine.utils import ProgressBar
 from scipy.optimize import differential_evolution
 from torch import Tensor
 
+from mmyolo.registry import DATASETS
 from mmyolo.utils import register_all_modules
 
 try:
@@ -602,8 +602,7 @@ def main():
     train_data_cfg = cfg.train_dataloader
     while 'dataset' in train_data_cfg:
         train_data_cfg = train_data_cfg['dataset']
-    # dataset = DATASETS.build(train_data_cfg)
-    dataset = build_dataset(train_data_cfg)
+    dataset = DATASETS.build(train_data_cfg)
 
     if args.algorithm == 'k-means':
         optimizer = YOLOKMeansAnchorOptimizer(