search

ros-industrial / easy_perception_deployment

A ROS2 package that accelerates the training and deployment of CV models for industries.
Apache License 2.0
53 stars 16 forks source link

Cannot generate dataset #26

Closed Tangetb closed 2 years ago

Tangetb commented 2 years ago

Creating dataset: ../data/datasets/custom_dataset/train_dataset Traceback (most recent call last): File "dataset/labelme2coco.py", line 182, in <module> main() File "dataset/labelme2coco.py", line 91, in main assert class_name == '__ignore__' AssertionError Creating dataset: ../data/datasets/custom_dataset/val_dataset Traceback (most recent call last): File "dataset/labelme2coco.py", line 182, in <module> main() File "dataset/labelme2coco.py", line 91, in main assert class_name == '__ignore__' AssertionError Have this error upon generating the dataset. I went to check the dataset file, the train_dataset and val_dataset have empty Jpeg folder and no annotation file is created

cardboardcode commented 2 years ago

Hi @Tangetb,

It seems the .txt label list file you have is missing the necessary __ignore__ class name.

Please refer to this section of the documentation for further clarification.

Your .txt label list file should look like this:

__ignore__
_background_
object_class_1_name
object_class_2_name