search

voldemortX / pytorch-auto-drive

PytorchAutoDrive: Segmentation models (ERFNet, ENet, DeepLab, FCN...) and Lane detection models (SCNN, RESA, LSTR, LaneATT, BézierLaneNet...) based on PyTorch with fast training, visualization, benchmarking & deployment help
BSD 3-Clause "New" or "Revised" License
840 stars 138 forks source link

如何二次开发 #168

Open Albertchamberlain opened 10 months ago

Albertchamberlain commented 10 months ago

比如我想在您的工作上添加一些模块，比如在/root/autodl-tmp/pytorch-auto-drive/configs/lane_detection/scnn/resnet18_tusimple.py 基础上进行，但是我并没有找到scnn的backbone在哪里，一些网络层，包括forward，您能告诉我该如何做吗？

voldemortX commented 10 months ago

@Albertchamberlain 你可以在整个文件夹里搜索config里面的类名

Albertchamberlain commented 10 months ago

谢谢，我去试试，

voldemortX commented 10 months ago

@Albertchamberlain 可以看一下这个文档：ADVANCED_TUTORIAL.md。比较符合你的需求。

Albertchamberlain commented 10 months ago

非常感谢