This is GenAI Image Classification project built with Kedro library.
We are trying to discriminate between images that are taken in the real world or created by hand and images generated using generative models like DALLE, Midjourney, or Stable Diffusion.
From our world:
AI generated:
Docker is required to run the project. To build the Docker image: Clone the repo and run the following command:
docker build -t deep-learning-project .
or just run:
docker pull kabanosk/deep-learning-project
To run the Docker container run the following command:
docker run -it --name dlp deep-learning-project
then go to this container
docker exec -it dlp bash
You can run Kedro project with:
kedro run
To visualize project run:
kedro viz run
You can run training pipeline using command:
kedro run --pipeline training
You can run evaluation pipeline using command:
kedro run --pipeline evaluation
To run API see: API
We welcome contributions! Please see our Contributing Guide for more details.
This project is licensed under the MIT License.
For any questions or concerns, please contact us at email.