Alibaba-MIIL / ASL

Official Pytorch Implementation of: "Asymmetric Loss For Multi-Label Classification"(ICCV, 2021) paper
MIT License
732 stars 102 forks source link

Update infer.py, add necessary import statement. #3

Closed yqtianust closed 4 years ago

yqtianust commented 4 years ago

The import for numpy and PIL Image is missing.

mrT23 commented 4 years ago

thanks for the corrections !