projet-long/README.md

sphereDetect

sphereDetect is a simple neural network, based on a Mask R-CNN, to detect spherical landmarks for image calibration.

Built with

• Python

Frameworks

• PyTorch
• TorchVision
• PyTorch Lightning
• PyTorch Metrics
• PyTorch Lightning Bolts
• ONNXRuntime

Tools

• Poetry
• Docker
• VSCode
  • ms-python
  • Python Docstring Generator
  • Conventional Commits
  • Remote container
  • EditorConfig
  • Docker

Getting started (with docker)

TODO

Getting started (without docker)

Installation

Clone the repository:

git clone git@git.inpt.fr:fainsil/pytorch-reva.git
cd pytorch-reva

Install the dependencies:

poetry install --with all

Usage

Activate python environment:

poetry shell

Start and configure Weights & Biases local server:

wandb server start
wandb login

Start a training:

python src/train.py

License

Distributed under the MIT license.
See LICENSE for more information.

Contact

Laurent Fainsin [fɛ̃zɛ̃]
<laurent@fainsin.bzh>