Mxnet迭代器从AWS S3获取数据

Question

如何修改此示例中的迭代器（https://github.com/dmlc/mxnet/blob/master/example/fcn-xs/data.py）以从AWS s3读取图像。我在AWS s3的文件夹中有.png图像。我尝试将rootdir作为s3：// bucketname /文件夹传递。我还尝试将从Image（通过PIL）读取图像的函数更改为imdecode（通过mx.image）。在这两种情况下我都没有运气。 我有图像分割问题。我的输入是一个图像，我的输出也是一个图像。

Answer 1

您似乎想要检索存储在s3中的单个图像。你可以使用python boto库。 我建议您修改line 56以从s3加载图片。

以下是一个例子：

    import StringIO
    from boto.s3.connection import S3Connection
    from PIL import Image

    aws_connection = S3Connection(AWS_KEY, AWS_SECRET)
    bucket = aws_connection.get_bucket(BUCKET_NAME)

    content = bucket.get_key(FILE_NAME).get_contents_as_string()
    image = Image.open(StringIO.StringIO(content))