Windows系统下tensorflow版本的YOLO v3

tensorflow版本的YOLO v3，在Windows系统下亲测可运行，代码完整，包含训练测试demo等等，上手简单容易，无需复杂的环境，

#! /usr/bin/env python
# coding=utf-8
#================================================================
#   Copyright （C） 2018 * Ltd. All rights reserved.
#
#   Editor      : VIM
#   File name   : convert_weight.py
#   Author      : YunYang1994
#   Created date: 2018-11-27 12:37:22
#   Description :
#
#================================================================

import os
import sys
# import wget
import time
import argparse
import tensorflow as tf
from core import yolov3 utils


class parser（argparse.ArgumentParser）:

    def __init__（selfdescription）:
        super（parser self）.__init__（description）

        self.add_argument（
            “--ckpt_file“ “-cf“ default=‘./scratches_checkpoint/yolov3.ckpt‘ type=str
            help=“[default: %（default）s] The checkpoint file ...“
            metavar=““
        ）

        self.add_argument（
            “--num_classes“ “-nc“ default=1 type=int
            help=“[default: %（default）s] The number of classes ...“
            metavar=““
        ）

        self.add_argument（
            “--anchors_path“ “-ap“ default=“./data/scratches_anchors.txt“ type=str
            help=“[default: %（default）s] The path of anchors ...“
            metavar=““
        ）

        self.add_argument（
            “--weights_path“ “-wp“ default=‘./checkpoint/yolov3.weights‘ type=str
            help=“[default: %（default）s] Download binary file with desired weights“
            metavar=““
        ）

        self.add_argument（
            “--convert“ “-cv“ action=‘store_true‘
            help=“[default: %（default）s] Downloading yolov3 weights and convert them“
        ）

        self.add_argument（
            “--freeze“ “-fz“ action=‘store_true‘
            help=“[default: %（default）s] freeze the yolov3 graph to pb ...“
        ）

        self.add_argument（
            “--image_h“ “-ih“ default=416 type=int
            help=“[default: %（default）s] The height of image 416 or 608“
            metavar=““
        ）

        self.add_argument（
            “--image_w“ “-iw“ default=416 type=int
            help=“[default: %（default）s] The width of image 416 or 608“
            metavar=““
        ）

        self.add_argument（
            “--iou_threshold“ “-it“ default=0.3 type=float
            help=“[default: %（default）s] The iou_threshold for gpu nms“
            metavar=““
        ）

        self.add_argument（
            “--score_threshold“ “-st“ default=0.3 type=float
            help=“[default: %（default）s] The score_threshold for gpu nms“
            metavar=““
        ）


def main（argv）:

    flags = parser（description=“freeze yolov3 graph from checkpoint file“）.parse_args（）
    print（“=> the input image size is [%d %d]“ %（flags.image_h flags.image_w））
    anchors = utils.get_anchors（flags.anchors_path flags.image_h flags.image_w）
    model = yolov3.yolov3（flags.num_classes anchors）

    with tf.Graph（）.as_default（） as graph:

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2019-03-08 21:18  scripts\
     文件        2928  2019-03-08 21:18  scripts\extract_coco.py
     文件        2337  2019-03-08 21:18  scripts\extract_voc.py
     文件         473  2019-03-08 21:18  scripts\make_raccoon_tfrecords.sh
     文件         883  2019-03-08 21:18  scripts\make_rbc_tfrecords.sh
     文件         863  2019-03-08 21:18  scripts\make_voc_tfrecords.sh
     文件         350  2019-03-08 21:18  .gitignore
     文件        5233  2019-03-24 11:05  convert_weight.py
     文件        6111  2019-03-25 19:46  evaluate.py
     文件       64959  2019-03-18 18:33  kmeans.jpg
     文件        5211  2019-03-18 18:35  kmeans.py
     文件        1064  2019-03-08 21:18  LICENSE
     文件        2696  2019-03-25 21:08  nms_demo.py
     文件        1357  2019-03-24 11:20  quick_test.py
     文件        5707  2019-03-28 21:59  quick_train.py
     文件        9086  2019-03-08 21:18  README.md
     文件        1753  2019-03-18 19:43  show_input_image.py
     文件        2245  2019-03-08 21:18  video_demo.py
     文件         262  2019-04-08 14:41  练习.py
     目录           0  2019-03-28 22:24  core\
     文件        1922  2019-03-17 19:39  core\common.py
     文件        2033  2019-03-18 19:22  core\convert_tfrecord.py
     文件       12369  2019-03-28 20:38  core\dataset.py
     文件         950  2018-01-20 00:55  core\timer.py
     文件       18631  2019-03-28 22:24  core\utils.py
     文件       18080  2019-03-28 21:46  core\yolov3.py
     目录           0  2019-03-25 21:07  core\__pycache__\
     文件        1408  2019-03-08 21:18  core\__pycache__\common.cpython-35.pyc
     文件        1369  2019-03-18 19:44  core\__pycache__\common.cpython-36.pyc
     文件        1283  2019-03-08 21:18  core\__pycache__\config.cpython-35.pyc
     文件        5632  2019-03-08 21:18  core\__pycache__\dataset.cpython-35.pyc
............此处省略712个文件信息