Skip to content

hdr-bgnn/BGNN-trait-segmentation

 
 

Repository files navigation

BGNN-trait-segmentation


PyTorch implementation of Fish trait segmentation model.

Requirements:


  • Python (>=3.6)
  • PyTorch
  • segmentation_models_pytorch
  • albumentations
  • Numpy
  • CV2
  • Matplotlib
  • Pillow
  • Seaborn
  • Pylab
  • Pandas

Source File


Run the single_model_training.ipynb notebook for training.