时间戳

首页 > 解决方案 > 如何添加个人数据增强？

tensorflow - 如何添加个人数据增强？

问题描述

我的代码如下所示：

 with strategy.scope():
        model_lib_v2.train_loop(
            pipeline_config_path=helper.pipeline_config_path,
            model_dir=model_dir,
            use_tpu=False,
            checkpoint_every_n=FLAGS.checkpoint_every_n,
            record_summaries=FLAGS.record_summaries)

所有数据扩充都添加到配置文件中，如下所示：

data_augmentation_options { random_distort_color { } } 那么，我该如何定义一个新的增强函数呢？

我的配置文件与https://github.com/tensorflow/models/blob/master/research/object_detection/samples/configs/ssd_resnet50_v1_fpn_shared_box_predictor_640x640_coco14_sync.config中的相同

现在，我想定义一个新的数据增强函数。我该怎么做？有教程吗？总的 tensorflow 代码太复杂了，我找不到在哪里添加函数。

标签： tensorflowtensorflow2.0object-detection

解决方案

推荐阅读