比较kmdoel和tflite推理输出

我的yolo3模型在k210里面输出结果完全不对,所以我十分怀疑是量化出了问题,但是我又找不到问题。还好昨天case小姐姐帮忙更新了nncase,可以在pc上推理kmdoel.然后我推理了几个图像,这次就是记录一下这个脚本,免得下次要用找不到了。。。

阅读全文

tf.keras自定义loss报错shape mismatch

在上次的文章中我写了如何自定义loss,但是我真正想要的使用的场景比那些还要复杂一些。

阅读全文

tf.dataset无法推断shape导致错误

使用tensorflow.keras的时候,tf.dataset在执行model.fit的时候报错:

ValueError: Cannot take the length of shape with unknown rank.

这里大概率是因为tf.dataset中使用了tf.py_function导致无法自动推导出张 良的形状,所以需要自己手动设置形状。

阅读全文

tf.keras多输出模型自定义loss

自从看了苏剑林的博客之后,我对keras是越来越喜欢了,但是我更喜欢在tensorflow中使用keras,今天就来看看如何在tf.keras中自定义多输出模型的loss,并且搭配高效的tf.dataset.

NOTE: tensorflow==2.0.0b0

阅读全文

numpy中动态范围切片

今天想把两个不同的形状的数组进行赋值,因为数组形状是动态的,所以要想一个办法进行动态的范围切片.

阅读全文

L softmx -> A softmx -> AM softmax

本篇文章是对Large Margin Softmax loss,Angular Margin to Softmax Loss,Additive Margin Softmax Loss的学习记录。公式我尽量按照原文来写,并加入一点注释。

阅读全文

tf.keras中分析性能

tensorflowkeras的结合太好了,今天训练模型超级慢,我准备使用tensorflow中的timeline来进行分析,一番寻找发现根本不需要以前那么麻烦。

阅读全文

tf.Keras完美使用tf.data API

最近看了苏剑林的几篇博客, 我忽然对keras不是那么抵触了,才发现之前认为Keras使用不灵活完全是因为的认识不够深入。所以我准备使用Tensorflow 2.0中的tf.Keras来 构建Yolo v3,在tensorflow中我们可以更加灵活的优化我们的数据输入管道,这次介绍一下多输入的model如何结合tf.data,基础的使用方式在这里学习。

阅读全文

变分自编码器(VAE)学习

我看了VAE之后忽然对神经网络的非监督学习以及概率模型很感兴趣,但是无奈概率模型真的好难懂啊. 今天尝试一边描述VAE一边真正的理解他. 参考总结自:https://spaces.ac.cn/archives/5253

阅读全文

python返回值进行unpack

最近在写yolov3,因为yolov3的多输出性质,所以我打算写适配多输出的工具函数,在numpy中可以在一个array中包含多个不同维度的array,但在tensorflow中一个tensor只能保存相同维度的矩阵,这就十分蛋疼了.下面记录一下我是如何解决的.

阅读全文