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.
@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