pytorch
CV伍六七
一只无厘头的CV算法工程师,一个不服输的斜杠青年,也是CV界的段子手。
小镇青年,双非出身,考研985,在卷到离谱的CV圈里成功上岸,目前正在万米裸泳中!冲冲冲!!!。
一边记录成长,一边分享收获。
展开
-
pytorch中stack和cat的以及to_tensor的坑
初入计算机视觉遇到的一些坑 1.pytorch中转tensor x=np.random.randint(10,100,(10,10,10)) x=TF.to_tensor(x) print(x) 这个函数会对输入数据进行自动归一化,比如有时候我们需要将0-255的图片转为numpy类型的数据,则会自动转为0-1之间 2.stack和cat之间的差别 stack x=torch.randn((1,...原创 2019-03-24 22:45:20 · 2686 阅读 · 0 评论 -
torchvison源码剖析【1】transform.ToTensor()
PIL转tensor: def pil2tensor(): img=Image.open(img_path) totensor=torchvision.transforms.ToTensor() #torchvision.transforms.ToTensor()函数自动转格式 img=totensor(img) img=img.cpu() prin...原创 2019-03-28 23:21:04 · 3966 阅读 · 0 评论