为TensorFlow / Keras创建对象检测数据集

时间:2019-03-06 17:24:00

标签: object-detection

我正在从模拟器中收集图像以进行目标检测。是否有用于标记每个图像的边界框(bx,by,bh,bw)并以TensorFlow格式生成它以用作(train_x,train_y)的软件? GUI应用程序会更好,因为它将使制作精确的边界框变得更加容易。

感谢您的帮助。谢谢

1 个答案:

答案 0 :(得分:1)

用于标记边界框的众所周知的工具是labelImg:https://github.com/tzutalin/labelImg

要将数据集转换为TensorFlow格式(我想您的意思是tfrecord格式),可以使用以下Python脚本:https://github.com/tensorflow/models/blob/master/research/object_detection/dataset_tools/create_pet_tf_record.py