search

zh320 / realtime-semantic-segmentation-pytorch

PyTorch implementation of over 30 realtime semantic segmentations models, e.g. BiSeNetv1, BiSeNetv2, CGNet, ContextNet, DABNet, DDRNet, EDANet, ENet, ERFNet, ESPNet, ESPNetv2, FastSCNN, ICNet, LEDNet, LinkNet, PP-LiteSeg, SegNet, ShelfNet, STDC, SwiftNet, and support knowledge distillation, distributed training etc.
Apache License 2.0
78 stars 14 forks source link

Enhancements #8

Open acai66 opened 3 months ago

acai66 commented 3 months ago
  1. Include custom dataset.
  2. Implement .gitignore file.
  3. Integrate requirements.txt.
  4. Ensure compatibility with Segformer backbone.
  5. Provide export onnx support for DDRNet and STDC.