Skip to content

Commit

Permalink
add kitti tracking vgg
Browse files Browse the repository at this point in the history
  • Loading branch information
Yu Xiang committed Jun 16, 2016
1 parent fb6a77e commit dcad990
Show file tree
Hide file tree
Showing 5 changed files with 1,210 additions and 0 deletions.
19 changes: 19 additions & 0 deletions fast-rcnn/experiments/cfgs/kitti_tracking_multiscale_vgg16.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
EXP_DIR: kitti_tracking
IS_RPN: False
IS_MULTISCALE: True
REGION_PROPOSAL: 'RPN'
TRAIN:
SCALES_BASE: !!python/tuple [3.0]
NUM_PER_OCTAVE: 1
IMS_PER_BATCH: 2
FG_FRACTION: 0.25
FG_THRESH: !!python/tuple [0.7, 0.5, 0.5]
BG_THRESH_HI: !!python/tuple [0.7, 0.5, 0.5]
BG_THRESH_LO: !!python/tuple [0.1, 0.1, 0.1]
BBOX_THRESH: !!python/tuple [0.7, 0.5, 0.5]
ROI_THRESHOLD: 0.01
SNAPSHOT_INFIX: kitti
TEST:
SCALES_BASE: !!python/tuple [3.0]
NUM_PER_OCTAVE: 1
NMS: 0.5
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
#!/bin/bash

set -x
set -e

export PYTHONUNBUFFERED="True"

LOG="experiments/logs/kitti_tracking_test_vgg16_rcnn_multiscale.txt.`date +'%Y-%m-%d_%H-%M-%S'`"
exec &> >(tee -a "$LOG")
echo Logging output to "$LOG"

time ./tools/train_net.py --gpu $1 \
--solver models/VGG16/kitti_tracking_test/solver_rcnn_multiscale.prototxt \
--weights data/imagenet_models/VGG16.v2.caffemodel \
--imdb kitti_tracking_training_trainval \
--cfg experiments/cfgs/kitti_tracking_multiscale_vgg16.yml \
--iters 80000

image_set="testing"

for i in {0..28}
do

seq_num=$(printf '%04d' "$i")
echo $seq_num

time ./tools/test_net.py --gpu $1 \
--def models/VGG16/kitti_tracking_test/test_rcnn_multiscale.prototxt \
--net output/kitti_tracking/kitti_tracking_training_trainval/vgg16_fast_rcnn_multiscale_trainval_kitti_iter_80000.caffemodel \
--imdb kitti_tracking_$image_set\_$seq_num \
--cfg experiments/cfgs/kitti_tracking_multiscale_vgg16.yml

# copy the detection result
cp output/kitti_tracking/kitti_tracking_$image_set\_$seq_num/vgg16_fast_rcnn_multiscale_trainval_kitti_iter_80000/$seq_num.txt data/KITTI_Tracking/detection_trainval_vgg16/$image_set

done
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
train_net: "models/VGG16/kitti_tracking_test/train_rcnn_multiscale.prototxt"
base_lr: 0.001
lr_policy: "step"
gamma: 0.1
stepsize: 60000
display: 20
average_loss: 100
momentum: 0.9
weight_decay: 0.0005
# We disable standard caffe solver snapshotting and implement our own snapshot
# function
snapshot: 0
# We still use the snapshot prefix, though
snapshot_prefix: "vgg16_fast_rcnn_multiscale_trainval"
#debug_info: true
Loading

0 comments on commit dcad990

Please sign in to comment.