Skip to content

A simple package to train and inference semantic segmentation models

Notifications You must be signed in to change notification settings

virasad/semantic_segmentation_service

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

84 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Semantic Segmentation Service

Train and inference semantic segmentation models easy as ABC.

How to use this

Set the variables in the docker-compose.yml file.

Train variables:

  • RESPONSE_URL: the url to send the training result
  • LOGGER_URL: the url to send the training log
  • IS_LOGGER_ON: enable logging or not

Inference variables:

  • WEIGHTS_DIR: the directory to read weights from it.

Run

docker compose up

For more details about the training please refer to the README.md file.

For more details about the inference please refer to the README.md file.

Docker Compose CI