Skip to content

a fork of vision transformer based framework for classifying executable images as benign or malicious

Notifications You must be signed in to change notification settings

thuvasooriya/vit-malware-detector

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

vit-malware-detector

refined vit train scripts for malware detection to test with risc-v cdac accelerator

changes

  • removed deprecated tfa (tensorflow_addons) in favor of keras
  • added requirements file for easy setup with python venv
  • integrated tensorflow-metal for faster local tests in macbook

todo

  • make sense of the code
  • host the trained model on huggingface to do inferences in google colab easily
  • more detailed output in terminal and in diagrams for diagonosis
  • make the script platform independant and human friendly

About

a fork of vision transformer based framework for classifying executable images as benign or malicious

Resources

Stars

Watchers

Forks

Languages

  • Python 100.0%