This package contains trained CNN models and methods dedicated to performs semantic segmentation of plant images.

git clone https://github.com/romi/romiseg

conda create -n romiseg python==3.10

python -m pip install 'torch>=2.0.0' 'torchvision>=0.15.0' --extra-index-url 'https://download.pytorch.org/whl/cu118'

We install it for CUDA 11.8, which is compatible with the latest PyTorch 2.x versions.

cd romiseg
python -m pip install -e .

• Segmentation2D module: outputs, prerequisites, and details are documented at docs/segmentation2d.md
• Annotation and fine‑tuning: step‑by‑step guide at docs/annotation-finetuning.md

For the full documentation website, see https://romi.github.io/romiseg/