wangzilu

@hangxu124 modify the cmakelist include_directories(${PROJECT_INCLUDE_DIR}/ here_is_your_caffe_path/include here_is_your_caffe_path/build/include)

@yfor1008 freeze_graph.freeze_graph(config['graphdef_file'], '',False, checkpoint_path,out_names, "save/restore_all", "save/Const:0", config['frozen_model_file'], False, "" ) ** 运行这一行的时候会有问题，之前存grapgh 的pb文件时候都可以。** Traceback (most recent call last): File "/home/wang/Downloads/tensorRT_for_keras-dev/train_densenet_tf.py", line 144, in "" File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/tools/freeze_graph.py", line 256, in freeze_graph...

还有一个问题就是我跑DenseNet不收敛，loss和acc基本没有变化，是加上scale层的情况下

我换版本到1.5以下以后FreezeGraph问题就解决了，但是问题是使用from tensorflow.python.layers.normalization import BatchNormalization加scale依旧acc不提高loss不下降 使用from tensorflow.python.keras._impl.keras.layers.normalization import BatchNormalization就可以

@yfor1008 感谢回复，我知道from tensorflow.python.keras._impl.keras.layers.normalization import BatchNormalization这个BN tensorRT不支持，我的问题是相同的超参， 使用这个BatchNorm loss 就是下降的，如果使用 tensorflow.python.layers.normalization import BatchNormalization就不会下降，后者确实可以将Frozen pb导出但是似乎acc基本没有变化，很困扰。