【问题标题】:TensorFLow Transfer Learning loading TFRecordDatasetTensorFlow 迁移学习加载 TFRecordDataset
【发布时间】:2020-12-15 01:18:08
【问题描述】:

我正在尝试按照迁移学习、Jupyter notebook、教程对马的图像进行分类:

https://github.com/tensorflow/docs/blob/master/site/en/tutorials/images/transfer_learning.ipynb

我正在尝试在 Docker 中构建本地 Tensorflow 训练管道,并安装了最新的“TensorFlow Docker Image”。

我有 1000 张带标签的 JPG 图片,800x800,带有边界框和相关的 annotations.coco.json;拆分为 train/validate/test 文件夹。

通过使用“create_coco_tf_record.py”脚本将“annotations.coco.json”转换为 TFRecords,我设法将图像和标签部分加载到 TFRecordDatasets。

我正在努力理解如何转换/调整这些数据的大小以输入到训练过程中,具体来说:

  • 如何在转换/调整大小中使用“边界框”数据?本教程将整个输入图像的大小调整为 160x160,没有要处理的“边界框”数据。

  • 例如,我是否应该剪掉包含在边界框内的图像,并调整它们的大小以输入到训练过程中?

  • 与“边界框”相关联,每张图像可以有多个边界框,因此每张图像有多个标签;我们如何处理每张图像的这些“多个标签”? 同样,本教程每张图片都有一个标签,因此在这方面更容易理解。

  • 是否有更合适的教程来处理“边界框”和“每个图像的多个标签”?

【问题讨论】:

    标签: python tensorflow jupyter-notebook transfer-learning image-classification


    【解决方案1】:

    我推荐这个 repo:https://github.com/zzh8829/yolov3-tf2 尝试微调 yolov3。它有一个完整的教程 (https://github.com/zzh8829/yolov3-tf2/blob/master/docs/training_voc.md) 关于如何重新训练这个 yolov3 对象检测器。它还支持“检测图像中的不同对象”(我认为=您的“每个图像有多个标签”)。希望这会让你继续前进,gl!

    【讨论】:

    • 感谢您的回复。
    • 将做一些阅读,看看它揭示了什么。我实际上已经设法加载带有边界框和多个标签的 TFRecords。需要锻炼过程中的后续步骤。
    猜你喜欢
    • 2020-10-31
    • 1970-01-01
    • 2021-08-16
    • 2020-08-06
    • 2022-11-27
    • 2016-05-18
    • 2021-06-03
    • 2016-11-28
    • 1970-01-01
    相关资源
    最近更新 更多