- 博客(4)
- 收藏
- 关注
原创 解决自定义batch_norm训练时报错:TypeError: cannot assign ‘torch.cuda.FloatTensor‘ as parameter ‘running_mean‘
其原因正如报错中所提到的,从batch_norm()函数中返回的是一个torch.cuda.FloatTensor类型的running_mean,不能直接赋值给Parameter类型,所以我们在此处可以做一下类型转换,即可消除这个报错,我训练下来应该没啥大问题,如果有错请大佬指正!我做了一些改动,将running_mean和running_var设置为了Parameter,如果直接赋值为tensor类型变量的话,是不会保存这两个参数的,会造成很多不便,但同时也不会报题目中的错误。
2023-07-18 15:33:25 701 1
原创 手势识别(肤色检测)重心计算以及重心变化趋势计算
代码的主体思想就是定义一个类似于FIFO的存储器,每隔十帧图像,会将一个重心数据存入存储器2中,同时存储器0使用存储器1的数据进行覆盖,存储器1使用存储器2的内容进行覆盖,这样就可以做到实时检测,而不用固定采集一定数量帧图像之后再进行检测。其中分子只剩下坐标的相加,分母其实代表图中白色像素块的个数,此时公式就比较利于硬件实现,而且利用分母的计算,我们可以进行一些肤色检测的抗干扰设计,例如。好了,本文到此就结束了,写这篇文章的主要目的还是记录一下几个星期以来的成果,大佬勿喷。(某些原因就不展示了QAQ)
2023-06-20 19:27:00 255 1
原创 HLS:csim由于内存不足而导致仿真失败——SIGSEGV
在其他博客中看到出现此错误最可能原因是数组访问越界,或者内存不足,经过检查之后(将其余全部代码注释,仅剩声明数组代码,仍然csim失败)访问越界被排除,以下是我设置的输入参数以及开辟的数组,实验发现一旦输入维度增大则cim直接失败。同时ap_uint所占用的空间大小为1字节,经过计算,在我设置的输入参数下,开辟的数组已经远远超过此范围,通过调整堆栈空间空间可解决此问题。将此贴作为错误记录。
2023-03-31 11:59:20 760 3
原创 仿真通过AXI_lite接口读写寄存器时axi_awready信号无法拉高的一种原因
仿真通过AXI_lite接口读写寄存器时axi_awready信号无法拉高的一种原因
2022-09-12 17:15:51 2787
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人