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

I can't run this code with dataset CamVid #6

Closed chris2000520 closed 3 months ago

chris2000520 commented 3 months ago

I wrote my own code to load the Camvid dataset, but when I use the model BisenetV2, it reports an error that the image size does not match. My image input size is crop_w = 480, crop_h = 360