Question

我正在使用对象检测API来训练不同的数据集，我想知道是否可以在训练期间获得到达网络的样本图像。

我问这个是因为我试图找到一个很好的数据扩充选项组合（here the options），但添加它们的结果更糟。看到培训中到达网络的内容将非常有帮助。

另一个问题是，如果传递的数据集使它们不平衡，是否有可能让API帮助平衡类。

谢谢！

Answer 1

是的，这是可能的。简而言之，您需要获取 tf.data.Dataset 的实例。然后，您可以迭代它并以 NumPy 数组的形式获取网络输入数据。然后使用 PIL 或 OpenCV 将其保存为图像文件是微不足道的。

假设您使用 TF2，伪代码如下：

ds = ... get dataset object somehow

sample_num = 0
for features, _ in ds:
    images = features[fields.InputDataFields.image]  # is a [batch_size, H, W, C] float32 tensor with preprocessed images
    batch_size = images.shape[0]
    for i in range(batch_size):
        image = np.array(images[i] * 255).astype(np.uint8)  # assuming input data is only scaled to [0..1]
        cv2.imwrite(output_path, image)

    sample_num += 1
    if sample_num >= MAX_SAMPLES:
        break

这里的技巧是获取 Dataset 实例。 Google 对象检测 API 非常复杂，但我想您应该从此处调用 train_input 函数开始：https://github.com/tensorflow/models/blob/3c8b6f1e17e230b68519fd8d58c4dd9e9570d789/research/object_detection/inputs.py#L763

它需要描述训练、train_input 和模型的管道配置子部分。

您可以在此处找到有关如何使用管道的一些代码片段：Dynamically Editing Pipeline Config for Tensorflow Object Detection

import argparse

import tensorflow as tf
from google.protobuf import text_format
from object_detection.protos import pipeline_pb2


def parse_arguments():                                                                                                                                                                                                                                                
    parser = argparse.ArgumentParser(description='')                                                                                                                                                                                                                  
    parser.add_argument('pipeline')                                                                                                                                                                                                                                   
    parser.add_argument('output')                                                                                                                                                                                                                                     
    return parser.parse_args()                                                                                                                                                                                                                                        


def main():                                                                                                                                                                                                                                                           
    args = parse_arguments()                                                                                                                                                                                                                                          
    pipeline_config = pipeline_pb2.TrainEvalPipelineConfig()                                                                                                                                                                                                          

    with tf.gfile.GFile(args.pipeline, "r") as f:                                                                                                                                                                                                                     
        proto_str = f.read()                                                                                                                                                                                                                                          
        text_format.Merge(proto_str, pipeline_config)

了解对象检测API

1 个答案: