PyTorch 2D Semantic Segmentation

Still too early for documentation...

Request for Features

  • Customizable image shape from arguments.
  • Cross-dataset mapping and testing.
  • Data augmentations in loaders.
  • Centralize dataset information in JSON files.
  • Dataset statistics module.
    • Add pixel heatmap to datasets statistics module.