Skip to content

Instantly share code, notes, and snippets.

@DrSleep
Last active August 16, 2020 17:04
Show Gist options
  • Save DrSleep/4bce37254c5900545e6b65f6a0858b9c to your computer and use it in GitHub Desktop.
Save DrSleep/4bce37254c5900545e6b65f6a0858b9c to your computer and use it in GitHub Desktop.
KITTI VISUAL ODOMETRY DATASET
## http://cvlibs.net/datasets/kitti/eval_semantics.php
## https://omnomnom.vision.rwth-aachen.de/data/rwth_kitti_semantics_dataset.zip
### DATASET FOR SEMANTIC SEGMENTATION
IMG_MEAN = np.array((104.00698793,116.66876762,122.67891434), dtype=np.float32)
BATCH_SIZE = 2
DATA_DIRECTORY = './../rwth_kitti_semantics_dataset/'
DATA_LIST_PATH = './../rwth_kitti_semantics_dataset/splits/train_tf.txt'
IGNORE_LABEL = 255
INPUT_SIZE = '321,321'
LEARNING_RATE = 1e-4
NUM_CLASSES = 14
NUM_STEPS = 20000
RANDOM_SEED = 1234
RESTORE_FROM = './deeplab_resnet.ckpt'
SAVE_NUM_IMAGES = 2
SAVE_PRED_EVERY = 10
SNAPSHOT_DIR = './snapshots_finetune/'
# colour map
label_colours = [(0, 0, 0),
(0, 0, 255),
(255, 0, 0),
(255,255, 0),
( 0,255, 0),
(255, 0,255),
( 0,255,255),
(255, 0,153),
(153, 0,255),
( 0,153,255),
(153,255, 0),
(255,153, 0),
( 0,255,153),
( 0,153,153)]
Display the source blob
Display the rendered blob
Raw
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@DrSleep
Copy link
Author

DrSleep commented Apr 28, 2017

Training results
kitti_1
kitti_2
kitti_3
kitti_4
kitti_5
kitti_6
kitti_7
kitti_8
kitti_9
kitti_10

@EternityZY
Copy link

good job!

@JadBatmobile
Copy link

what is the purpuse of IMG_MEAN in fine_tune.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment