以下のサイトの流れに沿ってkeras-yolo3の物体検知を試みています
SageMakerで「うまい棒検出モデル」を作ってみた
物体検出をするためにVottを使ってアノテーションを行い、jsonファイルを獲得しました。
今回はその後の学習、推論をAWSのAmazon SageMakerを使って行おうとしています。
そこでVottから得たjsonファイルをSageMakerで読み込めるフォーマットにする必要があるのですが、その変換コードがうまく動いてくれません
python3
1import json 2 3file_name = './vott-json-export/myfile-export.json' 4class_list = {'dog':0} 5 6with open(file_name) as f: 7 js = json.load(f) 8 9 for k, v in js["frames"].items(): 10 print(k, v) 11 k = int(k) 12 line = {} 13 line['file'] = '{0:04d}'.format(k+1) + '.jpg' 14 line['image_size'] = [{ 15 'width':int(v[0]['width']), 16 'height':int(v[0]['height']), 17 'depth':3 18 }] 19 20 line['annotations'] = [] 21 22 for annotation in v: 23 24 line['annotations'].append( 25 { 26 'class_id':class_list[annotation['tags'][0]], 27 'top':int(annotation['y1']), 28 'left':int(annotation['x1']), 29 'width':int(annotation['x2'])-int(annotation['x1']), 30 'height':int(annotation['y2']-int(annotation['y1'])) 31 } 32 ) 33 34 line['categories'] = [] 35 36 for name, class_id in class_list.items(): 37 38 line['categories'].append( 39 { 40 'class_id':class_id, 41 'name':name 42 } 43 ) 44 45 f = open('./json/'+'{0:04d}'.format(k+1) + '.json', 'w') 46 json.dump(line, f) 47 48
エラーとしては以下のものが出ます
Traceback (most recent call last): File "convert_json.py", line 9, in <module> for k, v in js["frames"].items(): KeyError: 'frames'
KeyErrorが出ている"frames"に関してなのですが、Vottでexportされたjsonファイルには記述がありませんでした。
解決法を知っている方、別の方法でSageMakerで物体検出する方法を知っている方ご教授よろしくお願いいたします
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。