tensorflow-addons.image 错误，输入错误？

Question

如何在tensorflow_addons.image 中使用translate？ 我尝试了以下没有成功

（版本：tensorflow: 2.4.1、tensorflow-addons: 0.12.1、python: 3.8.8）

import tensorflow as tf
import numpy as np
from tensorflow_addons.image import translate as tfa_translate

# 5 images 10x10, BW:
imgs = np.random.rand(5, 10, 10, 1).astype(np.float32)

# vector for x-y translations:
vec = np.random.randn(5, 2).astype(np.float32)

# both of these give me the same error below:
tfa_translate(imgs, vec)
tfa_translate(tf.convert_to_tensor(imgs), tf.convert_to_tensor(vec))

我得到的错误是关于使用 tf 或 np 张量。我尝试了tf.convert_to_tensor(imgs)，但没有成功：

NotImplementedError: Cannot convert a symbolic Tensor 
(translate/translations_to_projective_transforms/strided_slice:0) to a numpy array. 
This error` may indicate that you're trying to pass a Tensor to a NumPy call, 
which is not supported

编辑：这些examples 中的所有图像操作都在我的系统中工作。但翻译没有。

Answer 1

我能够以这种组合tensorflow: 2.4.1, tensorflow-addons: 0.12.1 and python: 3.7.10 执行上述代码。你会成功的。

import sys
import tensorflow as tf
import numpy as np
import tensorflow_addons as tfa
from tensorflow_addons.image import translate as tfa_translate
print(tf.__version__)
print(tfa.__version__)
print(sys.version)

# 5 images 10x10, BW:
imgs = np.random.rand(5, 10, 10, 1).astype(np.float32)

# vector for x-y translations:
vec = np.random.randn(5, 2).astype(np.float32)

# both of these give me the same error below:
tfa_translate(imgs, vec)
tfa_translate(tf.convert_to_tensor(imgs), tf.convert_to_tensor(vec))

输出：

2.4.1
0.12.1
3.7.10 
<tf.Tensor: shape=(5, 10, 10, 1), dtype=float32, numpy=
array([[[[0.73720914],
         [0.05389478],
         [0.00823868],
         [0.81953126],
         [0.1538573 ],
         [0.90768087],
         [0.4161973 ],
         [0.38890013],
         [0.7050291 ],
         [0.20925364]],

        [[0.37302   ],
         [0.4195738 ],
         [0.6881046 ],
         [0.3580131 ],
         [0.47772875],
         [0.5959794 ],
         [0.60560894],
         [0.03321437],
         [0.48405352],
         [0.7554338 ]],

        [[0.8335805 ],
         [0.69475925],
         [0.7356104 ],
         [0.72107244],
         [0.0654715 ],
         [0.7524717 ],
         [0.7242604 ],
         [0.8744024 ],
         [0.4255406 ],
         [0.8113107 ]],

        [[0.6129395 ],
         [0.29516342],
         [0.9028247 ],
         [0.13342927],
         [0.52949995],
         [0.06142981],
         [0.88151217],
         [0.52935964],
         [0.9567349 ],
         [0.8202645 ]],

        [[0.16741282],
         [0.6734363 ],
         [0.1131402 ],
         [0.8026573 ],
         [0.468824  ],
         [0.8553658 ],
         [0.45999104],
         [0.6134152 ],
         [0.56909144],
         [0.41677374]],

        [[0.35705274],
         [0.76314545],
         [0.07150301],
         [0.487003  ],
         [0.13564219],
         [0.45595258],
         [0.33955264],
         [0.9714719 ],
         [0.3500332 ],
         [0.6937466 ]],

        [[0.648089  ],
         [0.6019247 ],
         [0.3049946 ],
         [0.20571095],
         [0.2413046 ],
         [0.03899029],
         [0.85995656],
         [0.73361784],
         [0.83635384],
         [0.7748581 ]],

        [[0.4335603 ],
         [0.5851778 ],
         [0.11063848],
         [0.80494994],
         [0.49294096],
         [0.25171277],
         [0.81821585],
         [0.85311455],
         [0.6131411 ],
         [0.7952116 ]],

        [[0.7456568 ],
         [0.6693832 ],
         [0.05555756],
         [0.11654849],
         [0.674931  ],
         [0.01180321],
         [0.9755547 ],
         [0.4340065 ],
         [0.1364122 ],
         [0.31032914]],

        [[0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ]]],


       [[[0.34124446],
         [0.08647306],
         [0.9612431 ],
         [0.8725844 ],
         [0.6298909 ],
         [0.55960226],
         [0.3880017 ],
         [0.15042423],
         [0.58690244],
         [0.        ]],

        [[0.81984246],
         [0.8341217 ],
         [0.35999405],
         [0.3214622 ],
         [0.27512822],
         [0.905134  ],
         [0.5411199 ],
         [0.5287951 ],
         [0.56175965],
         [0.        ]],

        [[0.7129698 ],
         [0.17935541],
         [0.23184128],
         [0.62122566],
         [0.31914395],
         [0.93782455],
         [0.68015796],
         [0.32218182],
         [0.94558555],
         [0.        ]],

        [[0.9850416 ],
         [0.63286716],
         [0.02245921],
         [0.0168714 ],
         [0.787904  ],
         [0.7566337 ],
         [0.6966046 ],
         [0.74724126],
         [0.12862223],
         [0.        ]],

        [[0.9159414 ],
         [0.44754454],
         [0.4354201 ],
         [0.00121895],
         [0.9533174 ],
         [0.6897298 ],
         [0.5429657 ],
         [0.89288723],
         [0.17915004],
         [0.        ]],

        [[0.35772014],
         [0.25488794],
         [0.7412978 ],
         [0.77173555],
         [0.44523025],
         [0.515731  ],
         [0.6853389 ],
         [0.90855956],
         [0.28565803],
         [0.        ]],

        [[0.9447069 ],
         [0.25103524],
         [0.2835534 ],
         [0.5845979 ],
         [0.61925954],
         [0.5675663 ],
         [0.5097425 ],
         [0.7441513 ],
         [0.4252864 ],
         [0.        ]],

        [[0.71359545],
         [0.98984915],
         [0.25157735],
         [0.00311989],
         [0.59827286],
         [0.2282706 ],
         [0.74741423],
         [0.71861583],
         [0.4377698 ],
         [0.        ]],

        [[0.09316336],
         [0.90708685],
         [0.09424796],
         [0.49117166],
         [0.16317427],
         [0.8758143 ],
         [0.40267643],
         [0.9177538 ],
         [0.962702  ],
         [0.        ]],

        [[0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ]]],


       [[[0.        ],
         [0.        ],
         [0.29238865],
         [0.11071314],
         [0.04251722],
         [0.8379944 ],
         [0.7078936 ],
         [0.9524308 ],
         [0.9807888 ],
         [0.5980042 ]],

        [[0.        ],
         [0.        ],
         [0.6076964 ],
         [0.01608464],
         [0.5086389 ],
         [0.6162034 ],
         [0.13877843],
         [0.35256362],
         [0.49832624],
         [0.8281829 ]],

        [[0.        ],
         [0.        ],
         [0.16112986],
         [0.7567848 ],
         [0.3147346 ],
         [0.37325364],
         [0.7240583 ],
         [0.55212367],
         [0.53437054],
         [0.6456858 ]],

        [[0.        ],
         [0.        ],
         [0.9337465 ],
         [0.91347605],
         [0.20626037],
         [0.76980096],
         [0.42335644],
         [0.25908992],
         [0.11779127],
         [0.92509085]],

        [[0.        ],
         [0.        ],
         [0.42465472],
         [0.0791868 ],
         [0.260981  ],
         [0.98131573],
         [0.75192803],
         [0.94773096],
         [0.56423026],
         [0.41657394]],

        [[0.        ],
         [0.        ],
         [0.06342959],
         [0.14268413],
         [0.62750655],
         [0.40286288],
         [0.5470112 ],
         [0.16235335],
         [0.5923443 ],
         [0.2899487 ]],

        [[0.        ],
         [0.        ],
         [0.46837795],
         [0.64133584],
         [0.19043817],
         [0.3560451 ],
         [0.84368664],
         [0.54878306],
         [0.7572676 ],
         [0.63685185]],

        [[0.        ],
         [0.        ],
         [0.6362246 ],
         [0.9802078 ],
         [0.07812112],
         [0.6737748 ],
         [0.09005726],
         [0.5559371 ],
         [0.8273103 ],
         [0.8465655 ]],

        [[0.        ],
         [0.        ],
         [0.29354632],
         [0.8277752 ],
         [0.24557658],
         [0.7144999 ],
         [0.14932555],
         [0.7992488 ],
         [0.9322711 ],
         [0.7795503 ]],

        [[0.        ],
         [0.        ],
         [0.6565971 ],
         [0.9446715 ],
         [0.30254883],
         [0.87504804],
         [0.64962775],
         [0.13229366],
         [0.3159616 ],
         [0.80296475]]],


       [[[0.        ],
         [0.70018524],
         [0.7575729 ],
         [0.9112962 ],
         [0.7581045 ],
         [0.48528266],
         [0.7747405 ],
         [0.24268304],
         [0.51620704],
         [0.34689027]],

        [[0.        ],
         [0.95382404],
         [0.80364156],
         [0.48280716],
         [0.37804484],
         [0.24970461],
         [0.9051866 ],
         [0.7865404 ],
         [0.06981565],
         [0.44362214]],

        [[0.        ],
         [0.2775535 ],
         [0.9653021 ],
         [0.63792914],
         [0.7292578 ],
         [0.16679579],
         [0.43742415],
         [0.11834401],
         [0.79483396],
         [0.19500774]],

        [[0.        ],
         [0.07943237],
         [0.5328288 ],
         [0.77599657],
         [0.63793445],
         [0.19860882],
         [0.11559258],
         [0.9838874 ],
         [0.25520208],
         [0.5648425 ]],

        [[0.        ],
         [0.05485739],
         [0.68932915],
         [0.46699798],
         [0.42969576],
         [0.18972297],
         [0.46102318],
         [0.9518804 ],
         [0.49459723],
         [0.08304291]],

        [[0.        ],
         [0.3869838 ],
         [0.80616343],
         [0.29453617],
         [0.90575564],
         [0.9491408 ],
         [0.0916879 ],
         [0.36297712],
         [0.9152429 ],
         [0.67923504]],

        [[0.        ],
         [0.22770868],
         [0.04828448],
         [0.59498733],
         [0.8257741 ],
         [0.77833605],
         [0.30659854],
         [0.30935422],
         [0.96905756],
         [0.3825807 ]],

        [[0.        ],
         [0.2682056 ],
         [0.12400189],
         [0.14326625],
         [0.13485809],
         [0.60932964],
         [0.49995294],
         [0.48518673],
         [0.9061716 ],
         [0.6811738 ]],

        [[0.        ],
         [0.5348281 ],
         [0.4997965 ],
         [0.63221145],
         [0.46155205],
         [0.7663521 ],
         [0.91362387],
         [0.2826063 ],
         [0.68864125],
         [0.6800111 ]],

        [[0.        ],
         [0.39924976],
         [0.45846376],
         [0.7625721 ],
         [0.99027014],
         [0.39566088],
         [0.94737506],
         [0.4506151 ],
         [0.24510984],
         [0.89123327]]],


       [[[0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ],
         [0.        ]],

        [[0.        ],
         [0.9952434 ],
         [0.01445136],
         [0.20864646],
         [0.5734403 ],
         [0.9085302 ],
         [0.51775485],
         [0.8855587 ],
         [0.28934476],
         [0.06262866]],

        [[0.        ],
         [0.9793667 ],
         [0.5434468 ],
         [0.2679368 ],
         [0.10213093],
         [0.8542698 ],
         [0.2450935 ],
         [0.11804035],
         [0.8925573 ],
         [0.24042603]],

        [[0.        ],
         [0.4573032 ],
         [0.2561469 ],
         [0.00524052],
         [0.57438296],
         [0.5396885 ],
         [0.8146391 ],
         [0.9708599 ],
         [0.59276104],
         [0.35178396]],

        [[0.        ],
         [0.8042833 ],
         [0.7458325 ],
         [0.5945834 ],
         [0.9203529 ],
         [0.76014346],
         [0.69867104],
         [0.25335407],
         [0.96289504],
         [0.80579585]],

        [[0.        ],
         [0.16236259],
         [0.4947492 ],
         [0.41423783],
         [0.34527972],
         [0.4402823 ],
         [0.47581652],
         [0.82424086],
         [0.6161921 ],
         [0.6575118 ]],

        [[0.        ],
         [0.98747057],
         [0.66196424],
         [0.043115  ],
         [0.9993771 ],
         [0.12203664],
         [0.7465267 ],
         [0.3408288 ],
         [0.94268566],
         [0.8820024 ]],

        [[0.        ],
         [0.9850347 ],
         [0.5541368 ],
         [0.07462564],
         [0.07905959],
         [0.1890236 ],
         [0.6264842 ],
         [0.36688676],
         [0.6111452 ],
         [0.20477323]],

        [[0.        ],
         [0.07814959],
         [0.49131715],
         [0.14614591],
         [0.7088246 ],
         [0.77301365],
         [0.7900142 ],
         [0.6824042 ],
         [0.02469799],
         [0.20656963]],

        [[0.        ],
         [0.27418277],
         [0.2796882 ],
         [0.7702565 ],
         [0.9263017 ],
         [0.96371156],
         [0.9759365 ],
         [0.46440127],
         [0.12412024],
         [0.4896511 ]]]], dtype=float32)>