Open agnonchik opened 6 years ago
Maybe you could use the python file draw_net.py under the caffe/python/ to plot ssd architecture.
@Cathynuo Thanks for your suggestion. I know this tool. I just wonder whether amdegroot used draw_net.py to plot his SSD.jpg or some other tool?
I tried Caffe's draw_net.py on SSD-300 but the result had different look.
@agnonchik hi, you now can draw the ssd graph?
maybe need ourself code answer: https://github.com/amdegroot/ssd.pytorch/issues/52
I solved this problem,use https://github.com/szagoruyko/pytorchviz/blob/master/torchviz/dot.py
from torch.autograd import Variable
from utils.visualize import make_dot
from ssd import build_ssd
import torch
ssd_net = build_ssd('train', 512, 2)
net = ssd_net
x = Variable(torch.randn(1,3,512,512))
y = net(x)
dot = make_dot(y, params=dict(net.named_parameters()))
print(dot)
dot.view()
Which tool did you use to plot SSD.jpg? I want to replot it with higher resolution to make the text readable.