A PyTorch dataloader implementation for various image-based malware datasets. The package is available on PyPI.
| Dataset | Implementation Complete |
|---|---|
| MalImg (Google Drive) | ✅ |
| Microsoft Challenge (Kaggle) | ❌ |
| Mal-Net Tiny Image (Author Hosted Site) | ✅ |
| Mal-Net Full Image (Author Hosted Site) | ❌ |
| AdvNet: Lisa-CNN (Author Hosted Site | ✅* |
* We note that within the implementation of the AdvNet dataset, we have renamed all class label folders to Adv and Clean respectively.