Add 2d visualization feature

Fangyh09 / pytorch-receptive-field

Compute CNN receptive field size in pytorch in one line

Apache License 2.0

348 stars 57 forks source link

Closed nelson0515337 closed 10 months ago

nelson0515337 commented 10 months ago

[ ] Bug fix
[X] New feature
[ ] Breaking change
[ ] Refactor

I added a neat visualization function for 2D receptive field demonstration. Now, effortlessly highlight a model's receptive field on an image for clearer insights ✨

Fangyh09 commented 10 months ago

cool!