使用Sagemaker地面真相的图像中多个对象的边界框

时间:2018-12-19 10:50:53

标签: amazon-web-services amazon-sagemaker

我正在创建要在AWS Sagemaker object detection algo中使用的注释.json文件。格式如下:

{
   "file": "your_image_directory/sample_image1.jpg",
   "image_size": [
      {
         "width": 500,
         "height": 400,
         "depth": 3
      }
   ],
   "annotations": [
      {
         "class_id": 0,
         "left": 111,
         "top": 134,
         "width": 61,
         "height": 128
      },
      {
         "class_id": 0,
         "left": 161,
         "top": 250,
         "width": 79,
         "height": 143
      },
      {
         "class_id": 1,
         "left": 101,
         "top": 185,
         "width": 42,
         "height": 130
      }
   ],
   "categories": [
      {
         "class_id": 0,
         "name": "dog"
      },
      {
         "class_id": 1,
         "name": "cat"
      }
   ]
}

我想使用Sagemaker基本事实来获取对象坐标(使用边界框)以及class_id。有人可以提供一些建议,说明如何针对图像中的多个对象实现这一目标。我没有看到在地面真实情况下的边界框任务中提供多个标签(针对多个对象)的选项。

谢谢。

2 个答案:

答案 0 :(得分:0)

The place to start is the SM developers guide which links to a working aids for Ground Truth and explains the json format. The dg is here: https://docs.aws.amazon.com/sagemaker/latest/dg/sagemaker-dg.pdf.

For an example from aws of using sm for multi-object detection with SSD, try this: https://github.com/awslabs/amazon-sagemaker-examples/blob/master/introduction_to_amazon_algorithms/object_detection_pascalvoc_coco/object_detection_image_json_format.ipynb

答案 1 :(得分:0)

我是Ground Truth服务团队的成员。作为边框标记任务的一部分,我们不支持多个对象。我们认为这是一项功能请求,因此会优先考虑将来的版本。