Pytorch使用PIL和Numpy將單張圖片轉(zhuǎn)為Pytorch張量方式
我就廢話不多說了,大家還是直接看代碼吧!
#!/usr/bin/python # -*- coding: UTF-8 -*- from PIL import Image import numpy as np import torch pil_img = Image.open('/Users/songlu/Desktop/code.jpg') img = np.array(pil_img) print torch.from_numpy(img)
補充知識:pytorch mxnet 多GPU訓練 batch 的區(qū)別
區(qū)別有很多,這是一點點不一樣
1、mxnet 的batch ,設置的是單卡的,多卡是 batch*多卡,多一張卡,速度可以提升一倍
2、pytorch 是 batch/多卡數(shù)量,分配給每個gpu,設置的batch是總的batch
以上這篇Pytorch使用PIL和Numpy將單張圖片轉(zhuǎn)為Pytorch張量方式就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
python入門學習關(guān)于for else的特殊特性講解
本文將介紹 Python 中的" for-else"特性,并通過簡單的示例說明如何正確使用它,有需要的朋友可以借鑒參考下,希望能夠有所幫助2021-11-11一文帶你深入理解python中pytest-repeat插件的工作原理
這篇文章主要和大家一起深入探討到底pytest_repeat插件的具體功能是如何實現(xiàn)的呢,相信具體了解了該插件,其他三方插件也可以很快了解它內(nèi)部運行機制,所以本文詳細講解了python pytest-repeat插件的工作原理,需要的朋友可以參考下2023-09-09Python使用pickle模塊存儲數(shù)據(jù)報錯解決示例代碼
這篇文章主要介紹了Python使用pickle模塊存儲數(shù)據(jù)報錯解決示例代碼,小編覺得還是挺不錯的,具有一定借鑒價值,需要的朋友可以參考下2018-01-01