Thanks to visit codestin.com
Credit goes to github.com

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