Набор данных Tensorflow - карта json

Question

Я пытаюсь построить набор данных с API набора данных, моя ситуация следующая:

1. У меня есть набор json, указывающий для каждого экземпляра пару изображений (в моем изображение дела и его маска)
2. Я хочу отобразить каждый json и вернуть пару изображений в виде тензоров

В чистом python я бы сделал что-то вроде

import json
import tensorflow as tf
def parse_json(json_path):
  json_file = json.load(open(json_path))
  img = tf.io.read_file(json_file['img'])
  mask = tf.io.read_file(json_file['mask']

  return img, mask

Как я могу достичь того же результата с Tensorflow API? Спасибо за вашу помощь!