MzeroMiko / VMamba

VMamba: Visual State Space Models,code is based on mamba
MIT License
1.82k stars 98 forks source link

Custom Dataset #213

Open emrekeles-arch opened 1 month ago

emrekeles-arch commented 1 month ago

I want to train the model on custom data. This data consists of images and csv files containing annotations . What should be the structure of the file where the dataset will be located? Also, does the model work with csv files?

Thank you

MzeroMiko commented 3 weeks ago

You can refer to dataset for details. We just follow all the settings in swin-transformer.