ResNet model used for feature extraction

[mujavidb]

I can see from the paper that a ResNet18 model was used for feature extraction. On further examination of the pkl files, they are in a particular format, particularly the features which are a 1D array.

Do you know if more details are available on the model used to extract these features? Is it available to use by others or was it custom/fine-tuned?

[kylemin]

Hi,

Thank you for your interest in this work. Please refer to this link. You can use the checkpoints of ResNet-TSM models for the feature extraction.

Kyle

[charlotteburkirts]

Hi,

Just to follow up as I'm trying to train the model on a new dataset, in the pkl file you have a global_id for each detected face. Where does that come from? It is not in the original csv for the old repo SPELL neither.