dear author, thank for you provide code, i have some problem when i run your code in my own dataset.
"tensorflow.python.framework.errors_impl.InvalidArgumentError: No OpKernel was registered to support Op 'MeshSampling' with these attrs. Registered devices: [CPU], Registered kernels:
device='GPU‘"