search

uark-cviu / Micron-BERT

[CVPR 2023] Micron-BERT: BERT-based Facial Micro-Expression Recognition
126 stars 10 forks source link

micron_bert.py #2

Open ggg-s opened 1 year ago

ggg-s commented 1 year ago

get_model(checkpoint).cuda()-->get_model(checkpoint)[0].cuda() model.extract_features(image) ---> model.extract_features(image_tensor)