• 大小: 0.02M
    文件类型: .zip
    金币: 1
    下载: 0 次
    发布日期: 2021-03-27
  • 语言: 其他
  • 标签: 其他  

资源简介


将keras的.h5权重文件转换成darknet的weights,需要有模型结构文件

资源截图

代码片段和文件信息

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={‘anchors‘: anchors ‘num_classes‘: num_classes ‘

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           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.xml
     目录           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.xml
     文件         288  2018-12-11 14:58  h5_to_weight_yolo3-master\.idea\misc.xml
     文件         290  2018-12-11 14:58  h5_to_weight_yolo3-master\.idea\modules.xml
     文件         180  2018-12-11 14:58  h5_to_weight_yolo3-master\.idea\vcs.xml
     文件        9899  2019-02-01 23:20  h5_to_weight_yolo3-master\.idea\workspace.xml
     文件        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

评论

共有 条评论