之 keras-yolov3训练自己数据集第一部分

import json
from collections import defaultdict

name_box_id = defaultdict（list）
id_name = dict（）
f = open（
    “mscoco2017/annotations/instances_train2017.json“
    encoding=‘utf-8‘）
data = json.load（f）

annotations = data[‘annotations‘]
for ant in annotations:
    id = ant[‘image_id‘]
    name = ‘mscoco2017/train2017/%012d.jpg‘ % id
    cat = ant[‘category_id‘]

    if cat >= 1 and cat <= 11:
        cat = cat - 1
    elif cat >= 13 and cat <= 25:
        cat = cat - 2
    elif cat >= 27 and cat <= 28:
        cat = cat - 3
    elif cat >= 31 and cat <= 44:
        cat = cat - 5
    elif cat >= 46 and cat <= 65:
        cat = cat - 6
    elif cat == 67:
        cat = cat - 7
    elif cat == 70:
        cat = cat - 9
    elif cat >= 72 and cat <= 82:
        cat = cat - 10
    elif cat >= 84 and cat <= 90:
        cat = cat - 11

    name_box_id[name].append（[ant[‘bbox‘] cat]）

f = open（‘train.txt‘ ‘w‘）
for key in name_box_id.keys（）:
    f.write（key）
    box_infos = name_box_id[key]
    for info in box_infos:
        x_min = int（info[0][0]）
        y_min = int（info[0][1]）
        x_max = x_min + int（info[0][2]）
        y_max = y_min + int（info[0][3]）

        box_info = “ %d%d%d%d%d“ % （
            x_min y_min x_max y_max int（info[1]））
        f.write（box_info）
    f.write（‘
‘）
f.close（）

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2018-12-22 10:13  keras-yolo3-master
     文件        1201  2018-07-31 07:42  keras-yolo3-master.gitignore
     文件        1064  2018-07-31 07:42  keras-yolo3-masterLICENSE
     文件        1323  2018-07-31 07:42  keras-yolo3-mastercoco_annotation.py
     文件        5707  2018-07-31 07:42  keras-yolo3-masterdarknet53.cfg
     目录           0  2018-12-22 08:50  keras-yolo3-masterfont
     文件      127344  2018-07-31 07:42  keras-yolo3-masterfontFiraMono-Medium.otf
     文件        4432  2018-07-31 07:42  keras-yolo3-masterfontSIL Open Font License.txt
     文件        3482  2018-07-31 07:42  keras-yolo3-masterkmeans.py
     文件       37975  2018-12-22 08:57  keras-yolo3-master
esult.jpg
     文件       40124  2018-12-22 08:55  keras-yolo3-master est.jpg
     文件         859  2018-12-21 11:25  keras-yolo3-master est.py
     文件        2998  2018-12-21 11:45  keras-yolo3-master est.txt
     文件        4796  2018-12-21 12:22  keras-yolo3-master rain.py
     文件       30965  2018-12-21 11:45  keras-yolo3-master rain.txt
     文件       10686  2018-07-31 07:42  keras-yolo3-master rain_bottleneck.py
     文件         455  2018-12-21 11:45  keras-yolo3-masterval.txt
     文件        1245  2018-12-21 11:29  keras-yolo3-mastervoc_annotation.py
     目录           0  2018-12-22 08:50  keras-yolo3-masterwindows_v1.5.1
     目录           0  2018-12-22 08:50  keras-yolo3-masterwindows_v1.5.1data
     文件          14  2018-12-21 11:34  keras-yolo3-masterwindows_v1.5.1datapredefined_classes.txt
     文件    13259082  2017-09-27 19:16  keras-yolo3-masterwindows_v1.5.1labelImg.exe
     文件        8903  2018-12-22 08:53  keras-yolo3-masteryolo.py
     目录           0  2018-12-22 08:50  keras-yolo3-masteryolo3
     文件           0  2018-07-31 07:42  keras-yolo3-masteryolo3\__init__.py
     目录           0  2018-12-22 08:50  keras-yolo3-masteryolo3\__pycache__
     文件         145  2018-12-21 11:38  keras-yolo3-masteryolo3\__pycache__\__init__.cpython-36.pyc
     文件       12896  2018-12-21 11:38  keras-yolo3-masteryolo3\__pycache__model.cpython-36.pyc
     文件        3872  2018-12-21 11:38  keras-yolo3-masteryolo3\__pycache__utils.cpython-36.pyc
     文件       16491  2018-07-31 07:42  keras-yolo3-masteryolo3model.py
     文件        3843  2018-07-31 07:42  keras-yolo3-masteryolo3utils.py
............此处省略3个文件信息