search

peiyunh / tiny

Tiny Face Detector, CVPR 2017
https://cs.cmu.edu/~peiyunh/tiny
Other
1.13k stars 319 forks source link

how can I get the network description and visualization? #2

Closed sangthian closed 7 years ago

sangthian commented 7 years ago

Hello, I am very interested in your job. firstly, I want to to see what the networ look like, just like prototxt file in Caffe. Thanks.

peiyunh commented 7 years ago

After you load the net into matlab, you should be able to see its description as a structure array.

In terms of visualization, it gets a little bit trickier. Matconvnet has implemented model2dot.m, which takes a net and export a dot file (similar to how Caffe does it). Since our model contains some new (but very simple) layers, you may have to modify the original model2dot.m to make it work with our structure.

Cv9527 commented 7 years ago

would you release the training code for the academic goal?

peiyunh commented 7 years ago

I will release training code soon.