Finetuned google/vit-base-patch16-224 on Malimg Dataset contains 9,339 malware byteplot images from 25 different families. For iamge based malware classification.
Data Source: https://www.kaggle.com/datasets/ikrambenabd/malimg-original
logs : https://wandb.ai/prasad-cdac-work/vit-malware-classification-MalImg?nw=nwuserprasadcdacwork
Classification Report:
- Downloads last month
- 0
Inference Providers
NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API:
The HF Inference API does not support image-classification models for peft library.
Model tree for KakashiH/Malware_Image_Classifacation_ViT_base
Base model
google/vit-base-patch16-224