Skip to content

Instantly share code, notes, and snippets.

View ResNet-18-solver.prototxt
train_net: "models/apc/ResNet18/faster_rcnn_end2end/train.prototxt"
base_lr: 0.1
lr_policy: "step"
gamma: 0.1
stepsize: 50000
display: 20
average_loss: 100
# iter_size: 1
momentum: 0.9
weight_decay: 0.0005
@hgaiser
hgaiser / train.prototxt
Last active Sep 8, 2016
ResNet50-mnc3stage.prototxt
View train.prototxt
name: "ResNet50"
layer {
name: 'input-data'
type: 'Python'
top: 'data'
top: 'im_info'
top: 'gt_boxes'
top: 'gt_masks'
top: 'mask_info'
python_param {
@hgaiser
hgaiser / train.prototxt
Last active Sep 8, 2016
ResNet50 mnc5stage train
View train.prototxt
name: "ResNet50"
layer {
name: 'input-data'
type: 'Python'
top: 'data'
top: 'im_info'
top: 'gt_boxes'
top: 'gt_masks'
top: 'mask_info'
python_param {
@hgaiser
hgaiser / test.prototxt
Created Oct 5, 2016
ResNet50 light 3-stage
View test.prototxt
name: "ResNet50"
input: "data"
input_shape {
dim: 1
dim: 3
dim: 224
dim: 224
}
View simple_net.prototxt
name: "SimpleTriNet"
layer {
name: "data"
type: "Module"
top: "anchor"
top: "negative"
top: "positive"
module_param {
module: "triplet_layers"
View fkp.prototxt
name: "fkp_net"
layers {
name: "data"
type: MEMORY_DATA
top: "data"
top: "label"
memory_data_param {
batch_size: 1783 #batch size, so how many prediction youu want to do at once. Best is "1", but higher number get better performance
channels: 1
height: 96
View gist:5718a62ed3a5ebb2f92fc2a45ab43130
def get_session():
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
return tf.Session(config=config)
keras.backend.tensorflow_backend.set_session(get_session())
@hgaiser
hgaiser / mnist_tf.py
Last active Jul 13, 2017
MNIST example with Keras model and TF ops
View mnist_tf.py
#!/usr/bin/env python
import tensorflow as tf
from tensorflow.contrib.keras.api import keras
from tensorflow.contrib.keras.api.keras.models import Model, load_model
from tensorflow.contrib.keras.api.keras.layers import Input, Dense, Dropout, Flatten, Conv2D, MaxPooling2D, Activation, Lambda
from tensorflow.contrib.keras.api.keras.datasets import mnist
from tensorflow.contrib.keras.api.keras.utils import to_categorical
View ResNet50RetinaNet - Toy.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
View ResNet50RetinaNet - Toy.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.