pytorch
待何
这个作者很懒,什么都没留下…
展开
-
电脑GPU/CPU资源查看及使用
电脑GPU/CPU资源查看及使用import torch'''1. 服务器GPU状态查询 1) lspci | grep -i nvidia 可以查询所有nvidia显卡 2) lspci -v -s [显卡编号] 可以查看显卡具体属性 3) nvidia-smi 可以查看显卡的显存利用率2. torch.cuda主要函数 1) 查看是否有可用GPU、可用GPU数量: torch.cuda.is_available(), torch.cuda.device_原创 2020-08-28 14:56:37 · 2531 阅读 · 0 评论 -
遗传算法python代码(附详细注释)
遗传算法python代码(附详细注释)#代码参考:https://blog.csdn.net/ha_ha_ha233/article/details/91364937import numpy as np #用于数据操作:【X = np.linspace(*X_BOUND, 100) #将列表传入收集参数,完成解包】【 Y = np.linspace(*Y_BOUND, 100)】【X, Y = np.meshgrid(X, Y) #生成网格点坐标矩阵,每个二维矩阵表示一个维度】import mat原创 2020-08-28 14:53:28 · 9646 阅读 · 14 评论 -
python中[:] [::][-1][:-1][::-1]的意思 & np.arange与range的区别 &奇数偶数分离
python中[:] [::][:-1][::-1]的意思 & np.arange与range的区别python中的下标有两套,一套是正的(0,1,2,3,…),一套是负的(…,-3,-2,-1)。正的下标用0表示第一个元素,1表示第二个元素,依次类推;负的用-1表示最后一个元素,-2表示倒数第二个元素,依次类推。正负坐标可以同时使用。要点总结’ : ’ :表示遍历的意思,正向取值时使用’ :: ’ :表示遍历的意思,反向取值时使用[::-1]:表示反向取值b=a[ i : j :原创 2020-08-26 18:38:58 · 1579 阅读 · 0 评论 -
nohup代码理解
理解代码CUDA_VISIBLE_DEVICES=0 nohup python -u train.py --weight_path weight/yolov4.weights --gpu_id 0 > nohup.log 2>&1 &上述代码是下面代码的特例CUDA_VISIBLE_DEVICES=X nohup python -u 文件名.py --文件参数 > nohup.log 2>&1 &1 具体解释CUDA_VISIBLE_D原创 2020-08-25 21:17:27 · 787 阅读 · 1 评论 -
安装Apex Pycocotools(填坑之路)
安装Apex(NVIDA使用)在指定环境下输入以下代码git clone https://www.github.com/nvidia/apexcd apexpython setup.py install安装Pycocotools(coco数据集使用)在指定环境下输入以下代码git clone https://github.com/pdollar/coco.gitcd coco/PythonAPIpython setup.py build_ext --inplace原创 2020-08-25 15:50:16 · 666 阅读 · 0 评论 -
pytorch通过ImageFolder函数读取数据集(详细实例)
pytorch通过ImageFolder函数读取数据集(详细实例)dataset = ImageFolder(“E:/pycharmproject/dataset_read/veg200_images/”, transform=data_transform) #主要有两个参数,一个是图像根目录(被映射成标签的子目录的上一级),一个是数据操作本文理论参考,以下博客的方法二:https://blog.csdn.net/qq_36852276/article/details/945886561 代码#通过原创 2020-08-14 19:33:31 · 7527 阅读 · 7 评论 -
pytorch自建数据集并显示(详细示例)
pytorch自建数据集并显示本文建dataset类方法参考:https://blog.csdn.net/u011995719/article/details/851027701 代码加载数据集 path = ‘E:/pycharmproject/dataset_read/dataset_method_2/dataset_test.txt’ 按照自己具体情况修改import torchimport torchvision #用于下载数据集,进行图像增广操作等import matplotlib.p原创 2020-08-14 16:35:01 · 1212 阅读 · 0 评论 -
pytorch读取MNIST数据集并显示
pytorch读取MNIST数据集并显示代码:import torchimport torchvisionimport matplotlib.pyplot as plt #用于显示图片import osos.environ[“KMP_DUPLICATE_LIB_OK”]=“TRUE”#忽略警告import warningswarnings.filterwarnings(‘ignore’)#选择运行设备device = torch.device(‘cuda’ if torch.cuda.原创 2020-08-14 14:31:46 · 7201 阅读 · 0 评论