zhiyuanyou / SAFECount

[WACV 2023] Few-shot Object Counting with Similarity-Aware Feature Enhancement
Other
124 stars 22 forks source link

The train pipeline for CARPK dataset #2

Closed shanguangcv closed 1 year ago

shanguangcv commented 1 year ago

Hi, in cross-dataset (CARPK) experiment, you pre-train the model on FSC-147, can you provide a pre-train script? Moreover, can you also provide a fine-tune script for easy use? Thanks.

zhiyuanyou commented 1 year ago

Hi~

This needs some time to re-organize.

The main reason is that the settings of FSC147 & CARPK are different: for FSC147, exemplars are parts of the query image, for CARPK, exemplars are sampled from the training set and fixed for both training and evaluation.

We hope provide this in this week.

shanguangcv commented 1 year ago

Thanks

zhiyuanyou commented 1 year ago

Hi~ We have provided the pre-train and fine-tune script. If you have other questions, please re-open this issue.