architecture.py 文件源码

python
阅读 36 收藏 0 点赞 0 评论 0

项目:traffic_detection_yolo2 作者: wAuner 项目源码 文件源码
def build_model(inp_ph, train_flag_ph, lbl_ph, mask_ph):
    head = build_graph(inp_ph, train_flag_ph, 32, 5, 3)
    pred = yolo_prediction(head)
    loss = yolo_loss(lbl_ph, pred, mask_ph)

    return loss

#
# input_tensor = tf.placeholder(dtype=tf.float32, shape=(None, 416, 416, 3))
# train_flag = tf.placeholder(dtype=tf.bool)
# labels = tf.placeholder(tf.float32, shape=(None, 13, 13, 5, 7))
# mask = tf.placeholder(tf.bool, shape=(None, 13, 13, 5))
#
# head = build_graph(input_tensor, train_flag, 32, 5, 3)
# pred = yolo_prediction(head)
# loss = yolo_loss(labels, pred, mask)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号