Danilo is a Senior Software Engineer with an Associate Degree and 4 years of experience in some of the most famous IT Companies in Brazil.
Hi guys, a lot of people on the internet are looking for some way to analyze images and predict if it is sexual content or not (everyone by its own motivations). However, it's almost impossible to do it without thousands of images to train a convolutional neural network model. I'm making this article to show you that you can have a simple application that can do it for you, without worrying about neural networks stuff. We're going to use a convolutional neural network, but the model will be already trained, so you don't need to worry.
What am I going to learn?
- How to create a Python Rest API with Flask.
- How to create a simple service to check if the content is sexual or not.
- Docker Installed.
- Python 3 Installed.
- Pip Installed.
Creating the directory structure
- Open your favorite terminal.
- Create a project's root directory where we're going to put the project's files.
- Let's navigate to the folder we just created and create some files.
- Open the project's root directory with your favorite code editor.
Creating the Flask API
- Open the app.py file in your code editor.
- Let's code our prediction and health check routes.
Creating the Docker environment
- Let's implement our Dockerfile to install the required python modules and to run the application.
- Building the docker image.
- Starting a container on the port 80 of your local machine.
- The API should be running and ready to receive requests.
Testing our API
- Testing if the API is online. I'm using curl here, but you're free to use your favorite HTTP client.
- Expected response:
- Testing the classification route.
- Expected response:
- The score attribute in the response object is a guessing rate from 0 to 1, where 0 is equal to no sexual content, and 1 is equal to sexual content.
That's all folks! I hope you enjoyed this article, please let me know if you have some doubt.
You can get the source code of this article in the following link:
© 2019 Danilo Oliveira