Windows系统下tensorflow版本的YOLO v3

#! /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: