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
824 stars 137 forks source link

Heatmap代码 #163

Open Albertchamberlain opened 8 months ago

Albertchamberlain commented 8 months ago

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

voldemortX commented 8 months ago

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

Albertchamberlain commented 8 months ago

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

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

voldemortX commented 8 months ago

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

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

Albertchamberlain commented 8 months ago

谢谢