A web App for measuring the robustness of CNN Models relying on Adversarial Tests generated By Deepconcolic.
Overview • Technologies • Getting Started • How To Use • License
Romeo.AI is an application designed to evaluate the robustness of Convolutional Neural Network (CNN) classification models. This tool provides researchers, developers, and data scientists with a comprehensive and efficient way to analyze the reliability and performance of their CNN models under various scenarios.
With the increasing complexity of deep learning models and their applications in critical domains like autonomous vehicles, healthcare, and finance, it has become imperative to assess the robustness of these models against potential adversarial attacks, noisy data, and other challenging conditions. Romeo.AI empowers users to gain valuable insights into their CNN models' strengths and weaknesses, making it an essential tool in the machine learning development pipeline.
The core components of this software: - Python Flask Application - MongoDB Cloud Database - Docker Container
To get started with Romeo.AI, follow these simple steps:
- Install the application on your local machine or deploy it on a server accessible to your team.
- Prepare your pre-trained CNN classification model in a supported format (e.g., TensorFlow, PyTorch).
- Upload your model into Romeo.AI using the provided interface.
- Configure the robustness testing parameters based on your specific requirements.
- Initiate the robustness evaluation process and wait for the results to be generated. 6.Analyze the performance metrics and reports to gain insights into your model's robustness.
To clone and run this application, you'll need Git and Python Flask installed on your computer. From your command line:
# Clone this repository
$ git clone https://github.com/neirezcher/Romeo.AI.git
# Go into the repository
$ cd RomeoAI
# Activate your programming environment if you haven’t already
$ source env/bin/activate
# Once you have activated your programming environment, install dependencies using the pip install command:
$ pip install -r requirements.txt
# Set environment variables
$ export FLASK_APP= run.py
# Start Server
$ flask run
# or run this command
$ python -m flask run
NOTE: To run successfully this App you need to have Docker pre-installed on your machine.
MIT