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
852 stars 138 forks source link

Heatmap代码 #163

Open Albertchamberlain opened 1 year ago

Albertchamberlain commented 1 year ago

您好,能提供一下论文中绘画heatmap的代码吗?

voldemortX commented 1 year ago

@Albertchamberlain 可视化可能版本记录不全,复现度不高。我晚点找一下,应该给其他人发过。

Albertchamberlain commented 1 year ago

@Albertchamberlain 可视化可能版本记录不全,复现度不高。我晚点找一下,应该给其他人发过。

好的,谢谢,期待你的回复

voldemortX commented 1 year ago

@Albertchamberlain 去年我有一个邮件回复记录了相关代码,可视化有可能使用了resnet18的版本:

“关于可视化的code,当时是在重构前的PytorchAutoDrive (#45)的私有库做的。由于私有库有上百个分支,涉及一些要保密的内容,grad cam又没好好整理,不是很方便把全部代码给你(说实在的也不确定最终代码在哪个分支)。但重要部分大概当时是用的这个文件,我放到附件你可以结合去年旧的PytorchAutoDrive参考一下。” 7E06vA9.zip

Albertchamberlain commented 1 year ago

谢谢