Skip to content

Unofficial implementation of the EmotionROI essay and extended applications

License

Notifications You must be signed in to change notification settings

Kahsolt/EmotionROI-ex

Repository files navigation

EmotionROI-ex

Unofficial implementation of the EmotionROI essay and extended applications

ℹ The pretrained weights can be found here :)

Reproduction the FCNEL model in essay WHERE DO EMOTIONS COME FROM? PREDICTING THE EMOTION STIMULI MAP, and extended applications.

The results reported in the EmotionROI/FCNEL essay:

FCNEL

Reproduce FCNEL for ROI prediction

⚠ Can NOT reproduce essay results exactly, here're the implementation differences:

  • The essay uses FCN-32s + AlexNet, we use FCN-8s + ResNet50
  • The essay uses lr=1e-8 to refine 20 epochs, which is two small and doesn't have any effect in our setting, we instead use lr=1e-2 to train 100 epochs with CosineAnnealingLR
  • The essay might use the default Bilinear for output upsacling, we instead use Nearest-Exact to simulate FCN-32s resolution

FCNEL-infer
FCNEL-loss

Modifyied EmoFCNEL for ROI prediction with emotion condition

ℹ Predict ROI with any emotion label (except for sadness) results into nearly the same output...
ℹ The ROI annotations are probably NOT related to certain emotion, but only indicates human attention focus

EmoFCNEL-infer EmoFCNEL-loss EmoFCNEL-embed

Predict emotion directly from ROI annotation

⚠ Mission impossible :(

Model Train acc. Test acc.
Resnet50 24.098% 17.172%

RIO_clf

references


by Armit 2023/12/27

About

Unofficial implementation of the EmotionROI essay and extended applications

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages