nn_tools
nn_tools copied to clipboard
hahnyuan
Neural Network Tools: Converter and Analyzer. For caffe, pytorch, draknet and so on.
`Traceback (most recent call last): File "../nasnet-pytorch/pytorch_load.py", line 283, in pytorch_to_caffe.trans_net(model, dummy_input, name) File "/home/xiu/caffe-models-collection/nn_tools/pytorch_to_caffe.py", line 862, in trans_net out = net.forward(input_var) File "/home/xiu/caffe-models-collection/nasnet-pytorch/nasnet/nasnet.py", line 105, in forward x =...
I'm trying convert RefineDet to caffe model. RefineDet only contain regular CNN module(conv2d, convtranspose2d, maxpool, batchnorm). https://github.com/luuuyi/RefineDet.PyTorch However, pytorch2caffe failed at add operation as follow:  This add operation is...
$ python3 caffe_analyser.py slim-320.prototxt analys_result.csv 1,3,224,224 Traceback (most recent call last): File "caffe_analyser.py", line 28, in blob_dict, layers=profiling(net, Blob(shape)) File "/home/cli/桌面/nn_tools/analysis/CaffeA.py", line 15, in profiling input_name=net.net.input[0] IndexError: list index (0)...
I want to use your project to transform scnn pytorch model to caffe model，but it seems that the project doesn't work,the link of scnn model is as this "https://github.com/harryhan618/SCNN_Pytorch" would...
can you add the tool for scnn .pytorch to caffemodel~or can you give me some advice
Very great work, just find you code support the 0.3 version, and your implementation are very elegent, will you prepare to implement the Upsample operator? Thanks a lot
I have tried to change some source code, but the output caffemodel model has been verified to be incorrect. What do you think might go wrong? Please give me some...
Firstly, add the Sigmoid operation. Second, modified the interpolate. The original interpolate in the file uses Deconvolution, but it doesn't work rightly. So, I uses interp.cpp from Caffe to excute...
