ethereon / netscope

Neural network visualizer
http://ethereon.github.io/netscope
944 stars 322 forks source link

Some suggestion #13

Open huangh12 opened 7 years ago

huangh12 commented 7 years ago

It's an awesome work! It will be better if the label shape could also be infered as the image shape. At now, it is limited to [1], which only caters to the classification application. However, semantic segmentation is a hot topic today where the input label map has the same size as input image. So, the support to the shape infering of label map (not only the image/feature map) will make netscope even much better! :)