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

?屬性????????????大小?????日期????時(shí)間???名稱
-----------?---------??----------?-----??----
?????目錄???????????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??練習(xí).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個(gè)文件信息