torch div优化

举报
风吹稻花香 发表于 2021/06/04 23:20:42 2021/06/04
【摘要】 下面的方法比上面的快2倍左右,cpu占用差不多。 torch1.0好像没有这个优势了 python 11.5s torch能快一倍左右,6s import cv2if __name__ == '__main__':    path=r'D:\data\Original\JPEGImages\09981.jpg'     img_...

下面的方法比上面的快2倍左右,cpu占用差不多。

torch1.0好像没有这个优势了

python 11.5s

torch能快一倍左右,6s


  
  1. import cv2
  2. if __name__ == '__main__':
  3.     path=r'D:\data\Original\JPEGImages\09981.jpg'
  4.     img_ = cv2.imread(path)
  5.     start = time.time()
  6.     for j in range(500):
  7.         #python中如果是除法,img=img/255.0 会很快;img1=img/255.0 就会比较慢,一张图片需要156ms
  8.         img=img_/255.0
  9.         img2 = torch.from_numpy(img).float().unsqueeze(0)
  10.         # time.sleep(0.007)
  11.     print('time1',time.time()-start)
  12.     img = cv2.imread(path)
  13.     start = time.time()
  14.     for j in range(500):
  15.         # img_=img_/255.0
  16.         img_ = torch.from_numpy(img).float().div(255.0).unsqueeze(0)
  17.         # time.sleep(0.007)
  18.     print('time2', time.time() - start)

更快的方式:cuda,只需要2.5s

img = cv2.imread(path)
start = time.time()
for j in range(500): # img_=img_/255.0 img2 = torch.from_numpy(img) img2=img2.cuda().float().div(255.0).unsqueeze(0) # time.sleep(0.007)
print('time2', time.time() - start)

 

文章来源: blog.csdn.net,作者:网奇,版权归原作者所有,如需转载,请联系作者。

原文链接:blog.csdn.net/jacke121/article/details/84262114

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。