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, Optuna etc.
Apache License 2.0
133 stars 22 forks source link

Visualization code #10

Closed Shuo-Chen-2022 closed 6 months ago

Shuo-Chen-2022 commented 6 months ago

Thank you for your excellent work. Do you have any visualization code? I want to visualize my training results, thank you!

zh320 commented 6 months ago

Hi Shuo Chen, visualization codes already exist in this repo. You may turn on the flag _self.istesting within the config if you want to visualize your results.

Best