jankolf / grafiqs Goto Github PK

I set the run like that:
python extract_grafiqs.py --image-path /home/sm00th/Projects/assets/IMG_3625.jpg --output-dir /home/sm00th/Projects/apps/GraFIQs/output/ --backbone iresnet100 --weights /home/sm00th/Projects/apps/GraFIQs/backbones/resnet100_ms1mv2_arcface.pth

But got empty text files in the output dir...

Hi @jankolf,
I am using your method with another architecture which also have last layer returning 512 features. (below is a snapshot of last layer)

BatchNorm1d-543                  [-1, 512]           1,024
     ResNet-544                  [-1, 512]               0
================================================================
Total params: 47,181,073
Trainable params: 47,181,073
Non-trainable params: 0

However, the features I am getting after passing images through model have only one value of while the default model return tuple of 6 values. I am not able to figure out the reason for this mismatch. If you can explain the possible reason and ways to handle it, it will be very helpful.

Thanks
Girish