資源簡(jiǎn)介
將keras的.h5權(quán)重文件轉(zhuǎn)換成darknet的weights,需要有模型結(jié)構(gòu)文件
代碼片段和文件信息
from?keras.models?import?load_model
import?configparser
import?io
from?collections?import?defaultdict
from?keras.models?import?Model
import?numpy?as?np
from?yolo3.model?import?yolo_body?yolo_loss
from?keras.layers?import?Input?Lambda
num_anchors?=?9
num_classes?=?20
input_shape?=?(416416)
image_input?=?Input(shape=(None?None?3))
model_body?=?yolo_body(image_input?num_anchors//3?num_classes)
h?w?=?input_shape
y_true?=?[Input(shape=(h//{0:32?1:16?2:8}[l]?w//{0:32?1:16?2:8}[l]?\
????????num_anchors//3?num_classes+5))?for?l?in?range(3)]
def?get_anchors(anchors_path):
????‘‘‘loads?the?anchors?from?a?file‘‘‘
????with?open(anchors_path)?as?f:
????????anchors?=?f.readline()
????anchors?=?[float(x)?for?x?in?anchors.split(‘‘)]
????return?np.array(anchors).reshape(-1?2)
anchors_path?=?‘yolo_anchors.txt‘
anchors?=?get_anchors(anchors_path)
model_loss?=?Lambda(yolo_loss?output_shape=(1)?name=‘yolo_loss‘
????????arguments={‘a(chǎn)nchors‘:?anchors?‘num_classes‘:?num_classes?‘
?屬性????????????大小?????日期????時(shí)間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2019-02-01?23:25??h5_to_weight_yolo3-master\
?????目錄???????????0??2019-02-01?23:20??h5_to_weight_yolo3-master\.idea\
?????文件?????????453??2018-12-11?14:58??h5_to_weight_yolo3-master\.idea\convert_weight_file.iml
?????文件?????????138??2019-02-01?20:08??h5_to_weight_yolo3-master\.idea\encodings.xm
?????目錄???????????0??2018-12-11?14:58??h5_to_weight_yolo3-master\.idea\inspectionProfiles\
?????文件?????????809??2018-12-11?14:58??h5_to_weight_yolo3-master\.idea\inspectionProfiles\Project_Default.xm
?????文件?????????288??2018-12-11?14:58??h5_to_weight_yolo3-master\.idea\misc.xm
?????文件?????????290??2018-12-11?14:58??h5_to_weight_yolo3-master\.idea\modules.xm
?????文件?????????180??2018-12-11?14:58??h5_to_weight_yolo3-master\.idea\vcs.xm
?????文件????????9899??2019-02-01?23:20??h5_to_weight_yolo3-master\.idea\workspace.xm
?????文件????????6701??2019-02-01?21:09??h5_to_weight_yolo3-master\check_weight.py
?????文件??????????13??2018-12-11?14:58??h5_to_weight_yolo3-master\voc_classes.txt
?????文件????????8336??2018-12-11?14:58??h5_to_weight_yolo3-master\yolo.cfg
?????目錄???????????0??2019-02-01?21:07??h5_to_weight_yolo3-master\yolo3\
?????文件???????16491??2018-07-31?07:42??h5_to_weight_yolo3-master\yolo3\model.py
?????文件????????3843??2018-07-31?07:42??h5_to_weight_yolo3-master\yolo3\utils.py
?????目錄???????????0??2019-02-01?21:07??h5_to_weight_yolo3-master\yolo3\__pycache__\
?????文件???????12884??2019-02-01?21:06??h5_to_weight_yolo3-master\yolo3\__pycache__\model.cpython-36.pyc
?????文件????????3860??2019-02-01?21:07??h5_to_weight_yolo3-master\yolo3\__pycache__\utils.cpython-36.pyc
?????文件??????????76??2018-07-31?07:42??h5_to_weight_yolo3-master\yolo_anchors.txt
評(píng)論
共有 條評(píng)論