활연개랑

[Python (파이썬)] tfrecord 이미지 데이터 확인 본문

Python

[Python (파이썬)] tfrecord 이미지 데이터 확인

승해tmdhey 2022. 8. 21. 19:01
반응형
for image, mask in get_input_data(filenames).take(10):
    img = image.numpy().reshape(x, y, z)
    mask = mask.numpy().reshape(x, y, z)
    plt.figure(figsize=(12, 6))
    plt.subplot(121)
    plt.imshow(img[:, :, 10]) # 3d
    plt.title('img')
    plt.subplot(122)
    plt.imshow(mask[:, :, 10]) # 3d
    plt.title('mask')
    plt.show()