Error when converting VGG16_SOD from caffe to tensorflow about mmdnn

[kitstar]

Hi @seanchung2.

Fixed. Please update to newest code. Thanks very much.

Command:

$ python -m mmdnn.conversion._script.convertToIR -f caffe -d kit_imagenet -n deploy.prototxt -w VGG16_SOD_finetune.caffemodel

$ python -m mmdnn.conversion._script.IRToCode -f tensorflow -n kit_imagenet.pb -d kit_imagenet.py -w kit_imagenet.npy

$ python -m mmdnn.conversion.examples.tensorflow.imagenet_test -n kit_imagenet.py -w kit_imagenet.npy --dump ./vgg19_sod.ckpt

Tensorflow file is saved as [./vgg19_sod.ckpt], generated by [kit_imagenet.py] and [kit_imagenet.npy].

[namizzz]

Hi @Siddhant24
Now the way of dumping tensorflow model updates. You need to add "--dump_tag SERVING or TRAINING." .Here is README

[Siddhant24]

On running the last command I get this error

Traceback (most recent call last):
  File "/usr/lib/python3.6/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/usr/lib/python3.6/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/home/sid/MMdnn/mmdnn/conversion/examples/tensorflow/imagenet_test.py", line 106, in <module>
    raise ValueError("Need to provide the model type of Tensorflow model.")
ValueError: Need to provide the model type of Tensorflow model.

[Siddhant24]

@namizzz Yes that fixed the error. But instead of the messageTensorflow file is saved as [./vgg19_sod.ckpt], generated by [kit_imagenet.py] and [kit_imagenet.npy]., I get the following message Tensorflow file is saved as [b'vgg19_sod.ckpt/saved_model.pb'], generated by [kit_imagenet.py] and [kit_imagenet.npy].

[mtsol]

i cannot solve the error, and it is not solved at any of issues
File "c:\users\mba\anaconda3\lib\site-packages\mmdnn\conversion_script\conver
tToIR.py", line 89, in _convert
from mmdnn.conversion.pytorch.pytorch_parser import PytorchParser
File "c:\users\mba\anaconda3\lib\site-packages\mmdnn\conversion\pytorch\pytorc
h_parser.py", line 12, in
from mmdnn.conversion.pytorch.pytorch_graph import PytorchGraph
File "c:\users\mba\anaconda3\lib\site-packages\mmdnn\conversion\pytorch\pytorc
h_graph.py", line 12, in
from torch.jit import _unique_state_dict
ImportError: cannot import name '_unique_state_dict'