ImportError: cannot import name 'COCODataset' from 'yolox.data'

[kail85]

in yolox/exp/yolox_base.py, at below line

    def get_data_loader(self, batch_size, is_distributed, no_aug=False):
        from yolox.data import (
            COCODataset,
            TrainTransform,
            YoloBatchSampler,
            DataLoader,
            InfiniteSampler,
            MosaicDetection,
        )

Got an error 'ImportError: cannot import name 'COCODataset' from 'yolox.data''

Also tried

from yolox.data.datasets.coco import COCODataset

also got an error

cannot find module 'yolox.data.datasets.coco'.

Anyone please suggest? Thanks.

[CallmeAG]

I Got this error too. Do you solved it yet?

[FateScript]

I have tried it on my machine and it works.

In [1]: from yolox.data import COCODataset

In [2]: COCODataset?
Init signature: COCODataset(*args, **kwds)
Docstring:      COCO dataset class.
Init docstring:
COCO dataset initialization. Annotation data are read into memory by COCO API.
Args:
    data_dir (str): dataset root directory
    json_file (str): COCO json file name
    name (str): COCO data name (e.g. 'train2017' or 'val2017')
    img_size (int): target image size after pre-processing
    preproc: data augmentation strategy
File:           /data/workspace/YOLOX/yolox/data/datasets/coco.py
Type:           type
Subclasses:

Did you install the package through python3 -m pip install -v -e .?

[gyh420]

how to fix it?