AllentDan/LibtorchSegmentation

How to train on custom dataset without json file?

panovr opened this issue · 2 comments

panovr commented

Hi, my dataset has images and corresponding masks, but it has no json files. The dataset folders just like below:

Dataset
├── train
│   ├── 0.png
│   ├── 0_mask.png
│   └......
├── val
│   ├── 3.png
│   ├── 3_mask.png
│   └......

May I ask how to train on this dataset without json file?
Thanks!

AllentDan commented

Please modify get func here for your preference.

  • 👍1
antithing commented

@panovr did you successfully do this? Would you be able to share your code? I am looking to train with frames and alpha matte greyscale images.