- 博客(3)
- 收藏
- 关注
原创 yolov5的val.py输出的map值会随着batch_size的大小变化
在task的参数为’train, val, test时,会默认把图片填充成矩形形状,而在yolov5中,是把一个batch_size中所用的图像填充成相同大小,并填充的大小是根据一个batch_size中所用图像的大小计算的,这导致在batch_size不同时,同一张图像的填充大小不同,从而导致结果有一定的差距。
2023-04-21 22:46:29 547 1
原创 解决STN网络不收敛的方法
解决STN网络不收敛的方法方法: xs=self.localization(x) xs=xs.reshape(-1,32*14*2) theta=self.fc_loc(xs) theta=theta.view(-1,2,3) theta=torch.clamp(theta,-2,2)通过添加一个theta=torch.clamp(theta,-2,2),来验证theta的大小,防止图片经过变换后数据全部为零,造成网络不收
2021-07-16 01:56:31 201
原创 构建适合andriod的tensorflow lite
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档构建适合andriod的tensorflow lite前言一、环境配置二、使用步骤1、配置build2、运行常见问题前言在vm中安装ubun18.0.4进行编译,在tensorflow转化为tersorflow lite时,如果其中包含选择运算符要通过tensorflow的源代码构建tensorflow-lite.aar和tensorflow-lite-select-tf-ops.aar一、环境配置1、下载tensorflo
2021-07-14 17:04:19 388
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人