2

torch.Size([32, 3, 64, 64]) のような torch tensor があります。

私はテンソルをこれらのアサーションを渡すことができるものに変換しようとしています:

assert(type(images) == list)
assert(type(images[0]) == np.ndarray)
assert(len(images[0].shape) == 3)
assert(np.max(images[0]) > 10)
assert(np.min(images[0]) >= 0.0)

私は現在、テンソルを変換するためにこれを行っています:

# turn tensor into list of lists
imgs = imgs.tolist()

# iterate over list and turn each image into a numpy array with normalized values
for idx, img in enumerate(imgs):
  img = cv2.normalize(np.array(img), None,
  alpha = 0, beta = 255, norm_type = cv2.NORM_MINMAX )

そして、私はこのエラーを受け取ります:

File "scripts/run_model.py", line 158, in get_inception_score
assert(type(images[0]) == np.ndarray)
AssertionError

タイプ(画像)がリストでタイプ(画像[0]がnp.ndarray)になるようにテンソルを正しく変換するにはどうすればよいですか?どんな助けでも大歓迎です。前もって感謝します。

4

1 に答える 1