# MAINet: Multi-attribute IAA Framework

This code contains a simple example of how to use our MAINet. 
All information has been anonymized.


## Environmental Requirment

einops==0.3.0 \
matplotlib==2.0.2 \
numpy==1.13.1 \
opencv_python==4.2.0.34 \
Pillow==10.1.0 \
scipy==0.19.1 \
torch==1.9.0+cu111 \
torchvision==0.10.0+cu111 \
tqdm==4.66.1


## Run the Code

1. Download code from: "https://drive.google.com/drive/folders/1bhQgB07FA0X77lKJO5qsAEXGXsgpesJs?usp=sharing" and put it in the "code\" folder;
2. Downlod the backbone weights from "https://download.pytorch.org/models/resnet50-0676ba61.pth" and put it in the "code\model\" folder;
3. Download weights "iaa_aadb_holistic" or "iaa_aadb_attribute" from: "https://drive.google.com/drive/folders/1bhQgB07FA0X77lKJO5qsAEXGXsgpesJs?usp=sharing" and put it in the "code\IAA_weights\" folder;
4. Download datasets from their official website;
5. Run inference.py.
