changebio / AIC_Zeroshot

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AIC_Zeroshot

To do list

1, train an image classification model (***)

  • using some well-known structures training from scratch

resnet, mobilenet v2

2, train an object detection model (*)

3, image to attributes (**)

  • average attributes by top k prob

  • average attributes with prob wight by top k prob

  • remove fully connection layer of image classification model, finetune it to an image attributes model.

4, similarity between two categories by attribute metric (**)

About