Skip to content

Instantly share code, notes, and snippets.

View dongzhuoyao's full-sized avatar
💭
I may be slow to respond.

Tao Hu dongzhuoyao

💭
I may be slow to respond.
View GitHub Profile
name: "ResNet-50"
input: "data"
input_dim: 1
input_dim: 3
input_dim: 224
input_dim: 224
layer {
bottom: "data"
top: "conv1"
@dongzhuoyao
dongzhuoyao / gist:5d445456b49dc78e33d3afb5f61d9b33
Last active April 2, 2017 06:11
deeplabv2_vgg16_xavier_bn
# VGG 16-layer network convolutional finetuning
# Network modified to have smaller receptive field (128 pixels)
# nand smaller stride (8 pixels) when run in convolutional mode.
#
# In this model we also change max pooling size in the first 4 layers
# from 2 to 3 while retaining stride = 2
# which makes it easier to exactly align responses at different layers.
#
# For alignment to work, we set (we choose 32x so as to be able to evaluate
# the model for all different subsampling sizes):
name: "RCF"
layer {
name: "data"
type: "ImageLabelmapData"
top: "data"
top: "label"
include {
phase: TRAIN
}
layer {
name: "data"
type: "CPMData"
top: "data"
top: "label"
data_param {
source: "/home/zhecao/COCO_kpt/lmdb_trainVal"
batch_size: 10
backend: LMDB
}
input: "data"
input_dim: 1
input_dim: 3
input_dim: 368
input_dim: 368
layer {
name: "conv1_1"
type: "Convolution"
bottom: "data"
top: "conv1_1"
name: "RCF"
layer {
name: "data"
type: "ImageLabelmapData"
top: "data"
top: "label"
include {
phase: TRAIN
}
@dongzhuoyao
dongzhuoyao / KSVD.m
Created May 18, 2017 16:47 — forked from yangnk/KSVD.m
ksvd
function [Dictionary,output] = KSVD(...
Data,... % an nXN matrix that contins N signals (Y), each of dimension n.
param)
% =========================================================================
% K-SVD algorithm
% =========================================================================
% The K-SVD algorithm finds a dictionary for linear representation of
% signals. Given a set of signals, it searches for the best dictionary that
% can sparsely represent each signal. Detailed discussion on the algorithm
% and possible applications can be found in "The K-SVD: An Algorithm for
@dongzhuoyao
dongzhuoyao / xception
Created July 15, 2017 11:47
xception
name: "xception"
layer {
name: "data"
type: "Data"
top: "data"
top: "label"
transform_param {
mirror: false
crop_size: 224
mean_value: 104.0
name: "shufflenet"
# transform_param {
# scale: 0.017
# mirror: false
# crop_size: 224
# mean_value: [103.94,116.78,123.68]
# }
input: "data"
input_shape {
dim: 1
name: "xception"
layer {
name: "data"
type: "Data"
top: "data"
top: "label"
transform_param {
mirror: false
crop_size: 224
mean_value: 104.0