Я работаю над моделью Tensorflow Slim Resnet_v2. Моя версия тензорного потока (1.4.0) и версия Python (3.5.4). Я делаю классификацию изображений с двумя метками в качестве вывода. Для этой классификации я использую модель tensorflow Slim resnet_v2. Когда я пытаюсь запустить следующий код
predictions,_ = nets.resnet_v2.resnet_v2(x,num_classes=2, is_training=True,
global_pool=True,spatial_squeeze=True)
Я получаю следующую ошибку.
Traceback (most recent call last):
File "<ipython-input-2-a112bf21d73f>", line 1, in <module>
runfile('D:/Users/apxcm/Mammo_Transfer_learning/inception_fulltrain_azure.py', wdir='D:/Users/apxcm/Mammo_Transfer_learning')
File "D:\Users\apxcm\AppData\Local\Continuum\anaconda3\envs\nnet\lib\site-packages\spyder\utils\site\sitecustomize.py", line 710, in runfile
execfile(filename, namespace)
File "D:\Users\apxcm\AppData\Local\Continuum\anaconda3\envs\nnet\lib\site-packages\spyder\utils\site\sitecustomize.py", line 101, in execfile
exec(compile(f.read(), filename, 'exec'), namespace)
File "D:/Users/apxcm/Mammo_Transfer_learning/inception_fulltrain_azure.py", line 367, in <module>
main()
File "D:/Users/apxcm/Mammo_Transfer_learning/inception_fulltrain_azure.py", line 361, in main
resnet_full_classification()
File "D:/Users/apxcm/Mammo_Transfer_learning/inception_fulltrain_azure.py", line 326, in resnet_full_classification
predictions,_ = nets.resnet_v2.resnet_v2(x,num_classes=2, is_training=True,global_pool=True,spatial_squeeze=True)
TypeError: resnet_v2() got an unexpected keyword argument 'spatial_squeeze'
Я перепроверил, исходный код tensorflow slim (https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v2.py), а в строке 159 указано, что пространственное сжатие доступно. Я не знаю, почему я не могу использовать аргумент. Может ли это быть из-за версии тензорного потока? Если это так, пожалуйста, дайте мне знать, какую версию я должен использовать.