SmilingWolf/e621-v1-tfrecord

E621 TFRecords to train classifiers and other stuff with my codebases. TFRecord serialization/deserialization code: NUM_CLASSES = 8783 # Function to convert value to bytes_list def _bytes_feature(value): if isinstance(value, type(tf.constant(0))): value = value.numpy() elif isinstance(value, str): value = value.encode() return tf.train.Feature(bytes_list=tf.train.BytesList(value=[value])) # Function to convert bool/enum/int/uint to int64_list def… See the full description on the dataset page: https://huggingface.co/datasets/SmilingWolf/e621-v1-tfrecord.