search

Alibaba-MIIL / PETA

Official Pytorch Implementation of "PETA: Photo Albums Event Recognition using Transformers Attention" (2021)
18 stars 4 forks source link

Training script #14

Open justlike-prog opened 1 year ago

justlike-prog commented 1 year ago

Hi!

Thanks for the repo! Is it possible for you to share the training script? @tamarg1 or @ebenbaruch ?

Thanks!

tamarg1 commented 1 year ago

Sorry, none of us is at the company any longer so we don't have access. Are you trying to train the same use case or a different one?

justlike-prog commented 1 year ago

Oh ok, same usecase with different labels basically. @tamarg1

justlike-prog commented 1 year ago

Btw is the validation working correctly? Running it out of the box on the albums folder I get a map of 7.03. Using the AP_partial I get around 32 map. @tamarg1

justlike-prog commented 1 year ago

It seems that this mapping in utils.py isn't applied.

# Modify class idx for calculating mAP class_map = {0: "Birthday", 1: "Christmas", 2: "Graduation", 3: "Personal_sports", 4: "Show", 5: "ThemePark"}

What is the proper way to apply it?