python+tensorflow的yolo实现代码

yolo3的实现，平台是Windows10、python3.6.8 tensorflow1.12 运行yolo_video.py --camera 打开本地摄像头识别 运行yolo_video.py --image 打开图片识别 修改yolo.py中的代码，可以打开视频文件 缺少权重文件，可以自行下载weight文件用convert.py转换

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（）

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----

     文件       1201  2018-07-31 07:42  yolo3\.gitignore

     文件       1323  2018-07-31 07:42  yolo3\coco_annotation.py

     文件      10093  2018-07-31 07:42  yolo3\convert.py

     文件       5707  2018-07-31 07:42  yolo3\darknet53.cfg

     文件     127344  2018-07-31 07:42  yolo3\font\FiraMono-Medium.otf

     文件       4432  2018-07-31 07:42  yolo3\font\SIL Open Font License.txt

     文件       3482  2018-07-31 07:42  yolo3\kmeans.py

     文件       1064  2018-07-31 07:42  yolo3\LICENSE

     文件        625  2018-07-31 07:42  yolo3\model_data\coco_classes.txt

     文件         50  2018-07-31 07:42  yolo3\model_data\tiny_yolo_anchors.txt

     文件        135  2018-07-31 07:42  yolo3\model_data\voc_classes.txt

     文件         76  2018-07-31 07:42  yolo3\model_data\yolo_anchors.txt

     文件       4077  2018-07-31 07:42  yolo3\README.md

     文件       8332  2018-07-31 07:42  yolo3\train.py

     文件      10686  2018-07-31 07:42  yolo3\train_bottleneck.py

     文件       1416  2018-07-31 07:42  yolo3\voc_annotation.py

     文件       8592  2019-02-14 17:41  yolo3\yolo.py

     文件      16491  2018-07-31 07:42  yolo3\yolo3\model.py

     文件       3843  2018-07-31 07:42  yolo3\yolo3\utils.py

     文件          0  2018-07-31 07:42  yolo3\yolo3\__init__.py

     文件      12868  2019-02-14 10:28  yolo3\yolo3\__pycache__\model.cpython-36.pyc

     文件       3844  2019-02-14 10:28  yolo3\yolo3\__pycache__\utils.cpython-36.pyc

     文件        117  2019-02-14 10:28  yolo3\yolo3\__pycache__\__init__.cpython-36.pyc

     文件       8342  2018-07-31 07:42  yolo3\yolo3.cfg

     文件       1915  2018-07-31 07:42  yolo3\yolov3-tiny.cfg

     文件       2631  2019-02-19 14:14  yolo3\yolo_video.py

     文件       7137  2019-02-14 17:37  yolo3\__pycache__\yolo.cpython-36.pyc

     目录          0  2019-02-14 10:28  yolo3\yolo3\__pycache__

     目录          0  2019-02-14 10:12  yolo3\font

     目录          0  2019-02-19 14:21  yolo3\model_data

............此处省略6个文件信息