Skip to content

Cippppy/YMCA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

YOLO Multi-Object Color Attack (YMCA)

Created by Christian Cipolletta at Rowan University during the Spring 2024 Semester under Dr. Robi Polikar.

Description

In this study, we focus on designing a method to enhance the robustness and safety of computer vision systems, with a specific emphasis on YOLO models. Our objective is to assess the susceptibility of YOLO models to artificial coloring attacks and quantify their impact on model performance. Through the development of a methodology, we aim to identify and analyze the vulnerabilities of YOLO models to color-based perturbations. By doing so, we seek to contribute to the advancement of more resilient and reliable computer vision systems. We hypothesize that YOLO predictions are influenced by color variations, thereby suggesting a potential avenue for improving model robustness against adversarial attacks.

Table of Contents

  1. Installation
  2. Usage
  3. License
  4. Contact
  5. Appendix

Installation

To get started with this project, follow these steps:

  1. Clone the Repository: Begin by cloning this repository to your local machine using Git:

    git clone https://github.com/Cippppy/YMCA
  2. Navigate to the Project Directory: Change into the project directory:

    cd YMCA
  3. Install Dependencies: Install the required dependencies using pip. It's recommended to use a virtual environment to manage dependencies:

    pip install -r requirements.txt
  4. Download Data: Download the datasets required for this project from Google Drive. Follow the links below to access the datasets:

    Once downloaded, extract the datasets and place them in the datasets folder within the project directory. The "_colored" image folders are supposed to be empty.

  5. Ready to Use: You're all set! You can now proceed to use the project as described in the Usage section of this README.

If you encounter any issues during the installation process or have questions, feel free to reach out for assistance.

Usage

To use this project like we did in the paper all you need to do is use the command below:

python main.py

The scripts to create the figures and results from the paper are in the "results" directory. "plot.py" will create the figures. "runtime.py" will output the total runtime of the model's predictions. The total runtime is longer because of the coloring step.

python results/plot.py
python results/runtime.py

License

This project is licensed under the MIT License - see the LICENSE file for details.

The MIT License is a permissive open-source license that allows you to use, modify, and distribute the code for both commercial and non-commercial purposes. You are free to incorporate this project into your own work, subject to the terms and conditions outlined in the license file.

By using or contributing to this project, you agree to abide by the terms of the MIT License.

Contact

Feel free to reach out to me with any questions, feedback, or support requests!

Appendix: Additional Links

Here are some additional resources and links related to this project:

Feel free to explore these links for more information about the project.

About

YOLO Multi-Object Color Attack (YMCA) is an adversarial attack created by Christian Cipolletta as part of Rowan University's Engineering Clinic.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages