码迷,mamicode.com
首页 > 其他好文 > 详细

如何自动生成图片用于测试 pytorch(No image? No need image)

时间:2019-04-10 22:07:39      阅读:158      评论:0      收藏:0      [点我收藏+]

标签:oat   pre   code   nbsp   测试   image   如何   rand   class   

if __name__ == __main__:
    module = CAM_Module()
    in_data = torch.randint(0, 255, (2, 3, 7, 7), dtype=torch.float32)

 

如何自动生成图片用于测试 pytorch(No image? No need image)

标签:oat   pre   code   nbsp   测试   image   如何   rand   class   

原文地址:https://www.cnblogs.com/ziytong/p/10686055.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!