Malware Detection using Machine Learning

ISM project: Malware Detection using Machine Learning

Software Used

• Python 3.10 compiler
• Python Libraries:
  • pypng
  • python-image-library (PILlow)
  • tensorflow
  • tensorflow-gpu
• Visual Studio Code (Code Editor)
• Git

Usage

Clone this repository

git clone https://github.com/DarshK35/malware-detect-ml

Make sure to open a terminal in dist folder after cloning

Run exe-to-img for creating image dataset from executables

exe-to-img --folder [PATH_TO_SOURCE_FOLDER] --target [PATH_TO_TARGET_FOLDER] --count [COUNTER_VALUE_FOR_IMAGE_FILENAME] [--benign | --malware]

Sample output image is shown here. It was created by converting blender.exe executable

Run model