Python
nkszjx2
这个作者很懒,什么都没留下…
展开
-
windows 安装labelme
windows 安装labelme原创 2022-07-10 21:01:14 · 446 阅读 · 1 评论 -
计算CNN 模型算力(flops)和参数(parameters)数量计算
pip install ptflopspip install --upgrade git+https://github.com/sovrasov/flops-counter.pytorch.git代码如下:import torchvision.models as modelsimport torchfrom ptflops import get_model_complexity_infowith torch.cuda.device(0): net = models.16() .原创 2022-05-17 22:33:57 · 2144 阅读 · 1 评论 -
docker
问题描述pytorch 镜像位置:https://hub.docker.com/r/pytorch/pytorch/tags拉取镜像:docker pull pytorch/pytorch:1.5-cuda10.1-cudnn7-runtime1查看本地现有镜像清单:创建一个容器:docker run --gpus all -td --name=liujiepeng_pytorch pytorch/pytorch bash1注意:其实这里有两个repository名字相同,t原创 2022-04-23 18:25:12 · 1773 阅读 · 0 评论 -
Pytorch运行CNN代码读取数据出现StopIteration的异常及解决方案
如果只是单纯的依靠 next(trainloader_iter) 获取数据,则有可能获取数据时候出现不能获取的情况数据为non,此时我们需要再次回访该数据集,再次肚读取。代码如下:#读取数据代码部分: train_dataset = Dataset(args.data_dir, args.data_list, mean=IMG_MEAN) train_dataset_size = len(train_dataset) print ('dataset size:...原创 2022-04-07 16:33:20 · 2067 阅读 · 1 评论 -
Docker container 安装
sh build_dockerfile.shsh build_container.shdocker ps -alldocker start IDdocker attach ID具体细节如下:jianhua@ai4eo1:~/dockerImageConfig1$ docker ps -allCONTAINER ID IMAGE COMMAND CREATED ...原创 2021-11-15 21:44:46 · 1328 阅读 · 1 评论 -
python读取与保存Matlab的.mat文件
import scipy.io as scioimport pandas as pd# read *.matimport numpy as npdata_path="/Users/wc/Documents/MATLAB/data.mat"data_train = scio.loadmat(data_path)data_train_label=data_train.get('labe...翻译 2019-09-04 10:19:57 · 1173 阅读 · 0 评论