NVIDIA TAO 架构输入数据集 tfrecords 解决方案

第一用 coco 数据集转换
spec.txt 文件内容
coco_config {
root_directory_path: “/workspace/tao-experiments/data/coco”
img_dir_names: [“valid”]
annotation_files: [“valid/_annotations.coco.json”]
num_partitions: 1
num_shards: [1]
}

tao ssd dataset_convert -d /workspace/tao-experiments/data/coco/spec.txt -o /workspace/tao-experiments/data/coco/tfrecords/valid/valid