Skip to content

Instantly share code, notes, and snippets.

View arpan-dhatt's full-sized avatar

Arpan Dhatt arpan-dhatt

  • University of Texas at Austin
  • Austin, TX
  • 02:56 (UTC -05:00)
View GitHub Profile
@arpan-dhatt
arpan-dhatt / diode_depth.py
Last active January 2, 2020 03:22
Code for Diode Depth dataset for TFDS
import os
import numpy as np
import tensorflow as tf
import tensorflow_datasets.public_api as tfds
class DiodeDepth(tfds.core.GeneratorBasedBuilder):
"""Short description of my dataset."""
VERSION = tfds.core.Version('0.1.0')