How to train own dataset? about espnetv2 HOT 3 CLOSED

Training has two parts:
i) Data preparation: Here, you check if the data is in write format or not. Also, you compute class imbalance weights if your dataset is imbalanced. Please see loadData.py file for more details. It is pretty simple and accepts two text files: one for training and one for validation. You can create these files easily as these contains mapping between RGB image and Segmentation mask (ground truth) in comma separated form. Note that one line per image pair.

ii) Training: One you are done with this part i), you can start training your model on your dataset. You need to adjust the number of classes in the main.py file. Please see main.py file for more details. It should be straight forward.

Let me know if you are still encountering any issues.

from espnetv2.

I tried to train in the way you showed me.

However, I encountered such an error during training.

There are 12 dataset classes, and the size is 480 x 320.

I modified the code a little, but is there something missing?

from espnetv2.

I encountered the similar issue before. After I prepare the data image/annotation with 1024x512 resolution, the problem disappeared. I can't remember very clearly. You can have a try.

from espnetv2.