search

shamangary / FSA-Net

[CVPR19] FSA-Net: Learning Fine-Grained Structure Aggregation for Head Pose Estimation from a Single Image
Apache License 2.0
607 stars 153 forks source link

There is a problem when run demo_FSANET_ssd.py #53

Open dony1556 opened 4 years ago

dony1556 commented 4 years ago

Instructions for updating: dim is deprecated, use axis instead WARNING:tensorflow:From e:\sundwpyworkspace\vscode\FSA-Net\demo\lib\capsulelayers.py:195: calling softmax (from tensorflow.python.ops.nn_ops) with dim is deprecated and will be removed in a future version. Instructions for updating: dim is deprecated, use axis instead Traceback (most recent call last): File "e:/sundwpyworkspace/vscode/FSA-Net/demo/demo_FSANET_ssd.py", line 216, in main() File "e:/sundwpyworkspace/vscode/FSA-Net/demo/demo_FSANET_ssd.py", line 126, in main model1 = FSA_net_Capsule(image_size, num_classes, stage_num, lambda_d, S_set)() File "e:\sundwpyworkspace\vscode\FSA-Net\demo\lib\FSANET_model.py", line 444, in call ssr_Cap_list = ssr_aggregation_model(ssr_primcaps) File "D:\ProgramData\Anaconda3\envs\face_tf_sundw\lib\site-packages\keras\engine\topology.py", line 619, in call output = self.call(inputs, **kwargs) File "D:\ProgramData\Anaconda3\envs\face_tf_sundw\lib\site-packages\keras\engine\topology.py", line 2085, in call outputtensors, , _ = self.run_internal_graph(inputs, masks) File "D:\ProgramData\Anaconda3\envs\face_tf_sundw\lib\site-packages\keras\engine\topology.py", line 2289, in run_internal_graph assert str(id(x)) in tensor_map, 'Could not compute output ' + str(x) AssertionError: Could not compute output Tensor("reshape_8/Reshape:0", shape=(?, ?), dtype=float32)

PotentialPie commented 4 years ago

You can try to use keras 2.2.4 and tensorflow 1.10.0.