-
Notifications
You must be signed in to change notification settings - Fork 78
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Yu Xiang
committed
Jun 16, 2016
1 parent
fb6a77e
commit dcad990
Showing
5 changed files
with
1,210 additions
and
0 deletions.
There are no files selected for viewing
19 changes: 19 additions & 0 deletions
19
fast-rcnn/experiments/cfgs/kitti_tracking_multiscale_vgg16.yml
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
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 |
36 changes: 36 additions & 0 deletions
36
fast-rcnn/experiments/scripts/kitti_tracking_test_vgg16_rcnn_multiscale.sh
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
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 |
15 changes: 15 additions & 0 deletions
15
fast-rcnn/models/VGG16/kitti_tracking_test/solver_rcnn_multiscale.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
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 |
Oops, something went wrong.