├── .dockerignore ├── .gitignore ├── .pre-commit-config.yaml ├── .travis.yml ├── Dockerfile ├── README.md ├── app ├── __init__.py └── main.py ├── deep_learning_model ├── __init__.py ├── predictions │ ├── __init__.py │ └── classify_image.py ├── trained_model │ └── ResNet34model.pth └── training │ ├── __init__.py │ ├── config.py │ ├── models.py │ └── train.py ├── deployment-definition.yaml ├── pod-definition.yaml ├── replicaset-definition.yaml ├── requirements.dev.txt ├── requirements.txt ├── response_dto ├── __init__.py └── prediction_response_dto.py ├── sonar-project.properties ├── test_case_passed.PNG └── tests ├── __init__.py ├── assets ├── cat_image.jpg └── not_image.txt ├── conftest.py └── test_main.py /.dockerignore: -------------------------------------------------------------------------------- 1 | venv/ 2 | tests/ -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/.travis.yml -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/Dockerfile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/README.md -------------------------------------------------------------------------------- /app/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /app/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/app/main.py -------------------------------------------------------------------------------- /deep_learning_model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deep_learning_model/predictions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deep_learning_model/predictions/classify_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/deep_learning_model/predictions/classify_image.py -------------------------------------------------------------------------------- /deep_learning_model/trained_model/ResNet34model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/deep_learning_model/trained_model/ResNet34model.pth -------------------------------------------------------------------------------- /deep_learning_model/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /deep_learning_model/training/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/deep_learning_model/training/config.py -------------------------------------------------------------------------------- /deep_learning_model/training/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/deep_learning_model/training/models.py -------------------------------------------------------------------------------- /deep_learning_model/training/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/deep_learning_model/training/train.py -------------------------------------------------------------------------------- /deployment-definition.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/deployment-definition.yaml -------------------------------------------------------------------------------- /pod-definition.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/pod-definition.yaml -------------------------------------------------------------------------------- /replicaset-definition.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/replicaset-definition.yaml -------------------------------------------------------------------------------- /requirements.dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/requirements.dev.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/requirements.txt -------------------------------------------------------------------------------- /response_dto/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /response_dto/prediction_response_dto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/response_dto/prediction_response_dto.py -------------------------------------------------------------------------------- /sonar-project.properties: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/sonar-project.properties -------------------------------------------------------------------------------- /test_case_passed.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/test_case_passed.PNG -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/assets/cat_image.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/tests/assets/cat_image.jpg -------------------------------------------------------------------------------- /tests/assets/not_image.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/test_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jabertuhin/image-classification-api/HEAD/tests/test_main.py --------------------------------------------------------------------------------