Is it possible to train one class datasets? #2281
-
Hello, I only have one class (a simple cube) in my dataset I am asking myself, if it is even possible to train on a class that only got 1 class? The conversion with the collect_indoor3d_data.py seemed to work fine, Sadly i can not browse the data sets, neither S3DIS nor mine with tools/misc/browse_dataset.py since both are missing PALETTE Keyword. So maybe already the conversion failed. I tried my best in writing the necessary config files with those of S3DIS as template. But when training, the first thing the modell seems to try is: This way it achieves around 0.00644 in accuracy for the cube and 0 for clutter. The value does not change over time. I sorted my files in directories like they are used in S3DIS, instead of 6 Areas my data set has 24 Folders. In the first epoch i got -0,004 in loss, which seems wrong to me. Because training takes very long on these amount of data, and i wanted to change the parameters in the configs to find one that does anything i used a smaller set of images. NOTE: i needed to adjust the cross entropy function in the openmmlab/Lib/site-packages/torch/nn/ functional.py before the return line (arround line 3030) Which was suggested here: |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment
-
Here is an example of only training |
Beta Was this translation helpful? Give feedback.
Here is an example of only training
car
in kitti dataset: https://github.com/open-mmlab/mmdetection3d/blob/1.1/configs/_base_/datasets/kitti-3d-car.py#L7