YoloV5 代码BUG整理、技巧整理

本文介绍了在使用PyTorch时遇到的初始化库冲突问题及其解决方案,详细对比了FloatTensor和LongTensor的数据类型,并展示了index_select()、repeat方法的用法。同时,讨论了ImportError关于cv2模块的问题,以及try-except-else-finally语句的使用。最后,讲解了squeeze和unsqueeze在处理张量维度时的作用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1、Initializing libiomp5md.dll, but found libiomp5md.dll already initialized

用的链接中的情况一解决的,参考链接

2、torch.FloatTensor和torch.LongTensor的区别

torch.FloatTensor输出的是浮点型tensor数据,而torch.LongTensor输出的是整型tensor数据;

import torch
b=torch.FloatTensor(1,2,3)
c=torch.LongTensor(1,2,3)
print(b)
print(c)

输出:

tensor([[[0., 0., 0.],
         [0., 0., 0.]]])
tensor([[[0, 0, 0],
         [0, 0, 0]]])

b想转换成torch.LongTensor型,需通过.numpy()处理

b=torch.LongTensor(b.numpy())
print(b)

输出:

tensor([[[0, 0, 0],
         [0, 0, 0]]])

3、pytorch中index_select()的用法

参考:链接,解释的很清楚。

4、repeat方法

参考:链接,解释的也很清楚。

5、ImportError: cannot import name ‘_registerMatType‘ from ‘cv2.cv2‘

参考:链接

6、try-except-else-finally

参考:链接

7、squeeze和unsqueeze函数

参考:链接

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值