pytorch
Yao_tc
这个作者很懒,什么都没留下…
展开
-
pytorch ResNet 实现
from .BasicModule import BasicModulefrom torch import nnfrom torch.nn import functional as Fdef conv3x3(in_planes, out_planes, stride=1): """3x3 convolution with padding""" return nn.Conv...转载 2018-09-21 12:51:59 · 550 阅读 · 0 评论 -
写csv
# transfor csv formatdef tras(filename): import pandas as pd df = pd.read_csv(filename,header=None) df[0] = df[0].str.split('.jpg').str[0] df.columns = ['name', 'cls'] df['...原创 2018-09-21 14:18:58 · 311 阅读 · 0 评论 -
Docker for Pytorch 0.4.1+anconda3+cuda9.0+cudnn7+ubuntu16.04
docker fileFROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04MAINTAINER yaotc "yaotiechui@gmail.com"# RUN rm /etc/apt/sources.list.d/*# RUN rm /etc/apt/sources.list.d/cuda.list# COPY sources.list...原创 2018-10-16 16:49:37 · 2045 阅读 · 0 评论 -
Singularity配置Pytorch0.4.1 + Anaconda3(Python3.5) + CUDA9.0 + CUDNN7.0
Singularity recipeBootstrap: dockerFrom:nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04%post echo 'nameserver 8.8.8.8' >> /etc/resolv.conf apt-get update apt-get install -y wget ...原创 2018-10-30 15:05:45 · 1309 阅读 · 3 评论 -
NTU RGB+D Datasets
NTU RGB+D DatasetsBasic sizeThis dataset consists of 56,880 action samples containing 4 different modalities of data for each sample:RGB videos 136 GBdepth map sequencesMasked depth m...原创 2018-11-20 16:26:58 · 8750 阅读 · 10 评论 -
将caffe预训练模型的权重载入pytorch
#!/usr/bin/env python2.7#coding=utf-8import caffeimport csvimport numpy as np# np.set_printoptions(threshold='nan')MODEL_FILE = 'inception_v3_rgb_deploy.prototxt'PRETRAIN_FILE = 'inception_v3...原创 2018-12-04 15:51:11 · 3027 阅读 · 0 评论 -
attention
Created with Raphaël 2.2.0StartYour OperationYes or No?Endyesno原创 2019-01-09 12:46:29 · 234 阅读 · 0 评论 -
配置新服务器环境
serverEnvironmentDrivers: 390.87CUDA 9.0 , cudnn 7.3.0,OS: Ubuntu 16.04Pre-installsudo apt-get updatesudo apt-get install -y wget vim git cmake unzip unar openssh-serverAnaconda3 (Python3.6...原创 2019-04-04 18:18:34 · 494 阅读 · 0 评论