View ResNet-18-solver.prototxt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |
View train.prototxt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: "ResNet50" | |
layer { | |
name: 'input-data' | |
type: 'Python' | |
top: 'data' | |
top: 'im_info' | |
top: 'gt_boxes' | |
top: 'gt_masks' | |
top: 'mask_info' | |
python_param { |
View train.prototxt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: "ResNet50" | |
layer { | |
name: 'input-data' | |
type: 'Python' | |
top: 'data' | |
top: 'im_info' | |
top: 'gt_boxes' | |
top: 'gt_masks' | |
top: 'mask_info' | |
python_param { |
View test.prototxt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: "ResNet50" | |
input: "data" | |
input_shape { | |
dim: 1 | |
dim: 3 | |
dim: 224 | |
dim: 224 | |
} |
View simple_net.prototxt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: "SimpleTriNet" | |
layer { | |
name: "data" | |
type: "Module" | |
top: "anchor" | |
top: "negative" | |
top: "positive" | |
module_param { | |
module: "triplet_layers" |
View fkp.prototxt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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()) |
View mnist_tf.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/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.