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

该资源是我的博客《【YOLO初探】之 keras-yolov3训练自己数据集》代码的第一部分的内容。内容包括图像标记软件labelImg。加上第二、三部分的内容，就可以之家运行，请参看博客文章。

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（‘\n‘）
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-master\LICENSE
     文件        1323  2018-07-31 07:42  keras-yolo3-master\coco_annotation.py
     文件        5707  2018-07-31 07:42  keras-yolo3-master\darknet53.cfg
     目录           0  2018-12-22 08:50  keras-yolo3-master\font\
     文件      127344  2018-07-31 07:42  keras-yolo3-master\font\FiraMono-Medium.otf
     文件        4432  2018-07-31 07:42  keras-yolo3-master\font\SIL Open Font License.txt
     文件        3482  2018-07-31 07:42  keras-yolo3-master\kmeans.py
     文件       37975  2018-12-22 08:57  keras-yolo3-master\result.jpg
     文件       40124  2018-12-22 08:55  keras-yolo3-master\test.jpg
     文件         859  2018-12-21 11:25  keras-yolo3-master\test.py
     文件        2998  2018-12-21 11:45  keras-yolo3-master\test.txt
     文件        4796  2018-12-21 12:22  keras-yolo3-master\train.py
     文件       30965  2018-12-21 11:45  keras-yolo3-master\train.txt
     文件       10686  2018-07-31 07:42  keras-yolo3-master\train_bottleneck.py
     文件         455  2018-12-21 11:45  keras-yolo3-master\val.txt
     文件        1245  2018-12-21 11:29  keras-yolo3-master\voc_annotation.py
     目录           0  2018-12-22 08:50  keras-yolo3-master\windows_v1.5.1\
     目录           0  2018-12-22 08:50  keras-yolo3-master\windows_v1.5.1\data\
     文件          14  2018-12-21 11:34  keras-yolo3-master\windows_v1.5.1\data\predefined_classes.txt
     文件    13259082  2017-09-27 19:16  keras-yolo3-master\windows_v1.5.1\labelImg.exe
     文件        8903  2018-12-22 08:53  keras-yolo3-master\yolo.py
     目录           0  2018-12-22 08:50  keras-yolo3-master\yolo3\
     文件           0  2018-07-31 07:42  keras-yolo3-master\yolo3\__init__.py
     目录           0  2018-12-22 08:50  keras-yolo3-master\yolo3\__pycache__\
     文件         145  2018-12-21 11:38  keras-yolo3-master\yolo3\__pycache__\__init__.cpython-36.pyc
     文件       12896  2018-12-21 11:38  keras-yolo3-master\yolo3\__pycache__\model.cpython-36.pyc
     文件        3872  2018-12-21 11:38  keras-yolo3-master\yolo3\__pycache__\utils.cpython-36.pyc
     文件       16491  2018-07-31 07:42  keras-yolo3-master\yolo3\model.py
     文件        3843  2018-07-31 07:42  keras-yolo3-master\yolo3\utils.py
............此处省略3个文件信息