search

Verg-Avesta / CounTR

CounTR: Transformer-based Generalised Visual Counting
https://verg-avesta.github.io/CounTR_Webpage/
MIT License
92 stars 9 forks source link

Understanding PreTrain #9

Closed jaideep11061982 closed 1 year ago

jaideep11061982 commented 1 year ago

@Verg-Avesta Could you help understand the versions of pretain

Pretrain on FSC147 | models_mae_noct.py | FSC_pretrain.py -- | -- | -- Finetune on FSC147 | models_mae_cross.py | FSC_finetune_cross.py Finetune on CARPK | models_mae_cross.py | FSC_finetune_CARPK.py

What is difference between these files, Pretrain, finetunecross and finetune CARPK

Verg-Avesta commented 1 year ago

FSC_pretrain.py is used to pretrain the model on FSC147. FSC_finetune_cross.py is used to fine-tune the model on FSC147. FSC_finetune_CARPK.py is used to finetune the model on CARPK. Please read the paper for more details.