|
|
--- |
|
|
license: apache-2.0 |
|
|
language: |
|
|
- en |
|
|
metrics: |
|
|
- accuracy |
|
|
base_model: |
|
|
- google/vit-base-patch16-224 |
|
|
pipeline_tag: image-classification |
|
|
library_name: peft |
|
|
tags: |
|
|
- CyberSecurity |
|
|
- MalwareImageClassification |
|
|
- Malware |
|
|
--- |
|
|
|
|
|
## 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: |
|
|
|
|
|
 |
|
|
|
