![](https://img-blog.csdnimg.cn/20201014180756923.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
numpy
牛奶风车
这个作者很懒,什么都没留下…
展开
-
cv2或者numpy调换(图片)不同维度的顺序
如:HWC -> CHWimport cv2img = cv2.imread('1.jpg')img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)print(img.shape)img = img.transpose(2, 0, 1)print(img.shape)# 输出为:# (768, 1024, 3)# (3, 768, 1024)如果使用的是torch类型的:则可以使用permute函数.img = torch.from_numpy(原创 2021-07-27 11:20:38 · 3566 阅读 · 0 评论 -
简单图像卷积分类模型(CNN)
import torchimport torchvisionfrom torch.autograd import Variablefrom torchvision import datasets, transformsimport os # os包集成了一些对文件路径和目录进行操作的类import matplotlib.pyplot as pltimport time# 读取数据data_dir = 'Data' #Data 文件下有train 跟 valid两个转载 2021-07-20 22:19:45 · 480 阅读 · 0 评论 -
线性回归(一元曲线拟合 ) python
import numpy as npfrom sklearn.linear_model import LinearRegressionfrom sklearn.preprocessing import PolynomialFeaturesimport matplotlib.pyplot as pltx = np.array([1,1.1,1.2,1.3,1.4,1.5,1.6,1.7,1.8,1.9])# y = 0.8x^2 + 0.8x + 1 y = np.array([2.6,2.848原创 2021-07-18 10:28:35 · 543 阅读 · 0 评论