/the garage — 06 / 09

Itacatech Stage projectDevelopment

AI-Based Rock Recognition System

Computer vision for geological classification

Itacatech Stage project

AI-Based Rock Recognition System

Computer vision for geological classification

  • Java
  • Spring Boot
  • Python
  • Flask

/overview

An Itacatech stage project: a CNN classifies rock types from a photo, served through a Spring Boot API and a Flask microservice for prediction and metadata.

/approach

01

Model

Trained a CNN on labeled rock imagery for geological classification.

02

Serving

Wrapped the model in a Flask microservice dedicated to prediction and metadata extraction.

03

API layer

Built a Spring Boot layer in front of it, so classification could be consumed like any other backend endpoint.

/development

Designed and developed an AI system for rock type classification from images using a CNN, with a Spring Boot API layer and a Flask microservice for prediction and metadata extraction.
  • Java
  • Spring Boot
  • Python
  • Flask
  • TensorFlow
Itacatech Stage project

AI-Based Rock Recognition System

Computer vision for geological classification

  • Java
  • Spring Boot
  • Python
  • Flask