Open kail85 opened 3 years ago
I Got this error too. Do you solved it yet?
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 .
?
how to fix it?
in yolox/exp/yolox_base.py, at below line
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.