search

Js2Hou / Swin-CBIR

This repository contains a CBIR system that uses swin transformer to extract image's feature.
18 stars 1 forks source link

Training part #3

Closed saisusmitha closed 8 months ago

saisusmitha commented 1 year ago

Kindly let me know to train this. I mean if we want to apply for some other dataset.

Js2Hou commented 1 year ago

I just used swin-Transformer's pre-training checkpoint, where you can find the training details.

NQHuy1905 commented 1 year ago

@Js2Hou i see that, but how you tunned for Transformer with custom dataset, like other CNN we will freeze all layers except last layer and fine tuning it, but i don't see that option in their trainning file, how did you do?

Js2Hou commented 1 year ago

I didn't fine-tune the model.

NQHuy1905 commented 1 year ago

But is the model still good if i use that to extract feature with other classes not in ImageNet dataset

Js2Hou commented 1 year ago

Yes, swin-transformer has amazing performance