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 without smp #5

Closed chris2000520 closed 4 months ago

chris2000520 commented 4 months ago

I can't run main.py because it tells me I'm missing the smp module, but I don't want to use smp

zh320 commented 4 months ago

Hi, Chris, if you don't want to use smp, you can comment the following lines

https://github.com/zh320/realtime-semantic-segmentation-pytorch/blob/fc257a9a103080cb78bab2cf99c9869c41835a88/models/__init__.py#L2

https://github.com/zh320/realtime-semantic-segmentation-pytorch/blob/fc257a9a103080cb78bab2cf99c9869c41835a88/models/__init__.py#L41

chris2000520 commented 4 months ago

Hi, Chris, if you don't want to use smp, you can comment the following lines

https://github.com/zh320/realtime-semantic-segmentation-pytorch/blob/fc257a9a103080cb78bab2cf99c9869c41835a88/models/__init__.py#L2

https://github.com/zh320/realtime-semantic-segmentation-pytorch/blob/fc257a9a103080cb78bab2cf99c9869c41835a88/models/__init__.py#L41

oh！It works！Thank U！