search

Markin-Wang / FFVT

[BMVC 2021] The official PyTorch implementation of Feature Fusion Vision Transformer for Fine-Grained Visual Categorization
Other
48 stars 10 forks source link

Train with custom dataset #8

Open Fel26 opened 1 month ago

Fel26 commented 1 month ago

Hello, thank you for your awesome work. I want to ask about how can i train your model using custom dataset?

Thank you.

Markin-Wang commented 1 month ago

Thank you for your interest. To train our model using the custom dataset, the simpleset way is to define a class similar to https://github.com/Markin-Wang/FFVT/blob/9e44d1b0ce6eaf59af8f02202a527b4c352355bf/utils/dataset.py#L570 (only need to return the same thing as other datasets), then it should work. You may want to tune some hyper-paramteres to obtain the better results on your own dataset.