Skip to content

Commit

Permalink
bichen: fixed train/eval scripts. Added images for README.md
Browse files Browse the repository at this point in the history 
On branch master
Your branch is up-to-date with 'origin/master'.

Changes to be committed:
  new file:   README/det_img.png
  new file:   README/detection_analysis.png
  new file:   README/graph.png
  renamed:    data/out/out_sample.png -> README/out_sample.png
  modified:   scripts/eval_train.sh
  modified:   scripts/eval_val.sh
  modified:   scripts/train.sh
  • Loading branch information
@BichenWuUCB
BichenWuUCB committed Dec 10, 2016
1 parent 90ad4f6 commit 91ee8b4
Show file tree
Hide file tree
Showing 7 changed files with 29 additions and 29 deletions.
Binary file added README/det_img.png
View file
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added README/detection_analysis.png
View file
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added README/graph.png
View file
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
0 data/out/out_sample.png → README/out_sample.png
View file
File renamed without changes
8 changes: 4 additions & 4 deletions scripts/eval_train.sh
View file
Expand Up @@ -8,7 +8,7 @@ python ./src/eval.py \
--data_path=./data/KITTI \
--image_set=train \
--eval_dir=/tmp/bichen/logs/SqueezeDet/eval_train \
--checkpoint_dir=./data/model_checkpoints/squeezeDet/ \
--checkpoint_dir=/tmp/bichen/logs/SqueezeDet/train \
--net=squeezeDet \
--gpu=0

Expand All @@ -20,7 +20,7 @@ python ./src/eval.py \
# --data_path=./data/KITTI \
# --image_set=train \
# --eval_dir=/tmp/bichen/logs/SqueezeDetPlus/eval_train \
# --checkpoint_dir=./data/model_checkpoints/squeezeDetPlus/ \
# --checkpoint_dir=/tmp/bichen/logs/SqueezeDetPlus/train \
# --net=squeezeDet+ \
# --gpu=0

Expand All @@ -32,7 +32,7 @@ python ./src/eval.py \
# --data_path=./data/KITTI \
# --image_set=train \
# --eval_dir=/tmp/bichen/logs/vgg16/eval_train \
# --checkpoint_dir=./data/model_checkpoints/vgg16/ \
# --checkpoint_dir=/tmp/bichen/logs/vgg16/train \
# --net=vgg16 \
# --gpu=0

Expand All @@ -44,6 +44,6 @@ python ./src/eval.py \
# --data_path=./data/KITTI \
# --image_set=train \
# --eval_dir=/tmp/bichen/logs/resnet50/eval_train \
# --checkpoint_dir=./data/model_checkpoints/resnet50/ \
# --checkpoint_dir=/tmp/bichen/logs/resnet50/train \
# --net=resnet50 \
# --gpu=0
8 changes: 4 additions & 4 deletions scripts/eval_val.sh
View file
Expand Up @@ -8,7 +8,7 @@ python ./src/eval.py \
--data_path=./data/KITTI \
--image_set=val \
--eval_dir=/tmp/bichen/logs/SqueezeDet/eval_val \
--checkpoint_dir=./data/model_checkpoints/squeezeDet/ \
--checkpoint_dir=/tmp/bichen/logs/SqueezeDet/train \
--net=squeezeDet \
--gpu=0

Expand All @@ -20,7 +20,7 @@ python ./src/eval.py \
# --data_path=./data/KITTI \
# --image_set=val \
# --eval_dir=/tmp/bichen/logs/SqueezeDetPlus/eval_val \
# --checkpoint_dir=./data/model_checkpoints/squeezeDetPlus/ \
# --checkpoint_dir=/tmp/bichen/logs/SqueezeDetPlus/train \
# --net=squeezeDet+ \
# --gpu=0

Expand All @@ -32,7 +32,7 @@ python ./src/eval.py \
# --data_path=./data/KITTI \
# --image_set=val \
# --eval_dir=/tmp/bichen/logs/vgg16/eval_val \
# --checkpoint_dir=./data/model_checkpoints/vgg16/ \
# --checkpoint_dir=/tmp/bichen/logs/vgg16/train \
# --net=squeezeDet+ \
# --gpu=0

Expand All @@ -44,6 +44,6 @@ python ./src/eval.py \
# --data_path=./data/KITTI \
# --image_set=val \
# --eval_dir=/tmp/bichen/logs/resnet50/eval_train \
# --checkpoint_dir=./data/model_checkpoints/resnet50/ \
# --checkpoint_dir=/tmp/bichen/logs/resnet50/train \
# --net=resnet50 \
# --gpu=0
42 changes: 21 additions & 21 deletions scripts/train.sh
View file
Expand Up @@ -3,23 +3,23 @@
# =========================================================================== #
# command for squeezeDet:
# =========================================================================== #
# python ./src/train.py \
# --dataset=KITTI \
# --pretrained_model_path=./data/squeezenet/squeezenet_pretrained.pkl \
# --data_path=./data/KITTI \
# --image_set=train \
# --train_dir=/tmp/bichen/logs/SqueezeDet/train \
# --net=squeezeDet \
# --summary_step=100 \
# --checkpoint_step=500 \
# --gpu=0
python ./src/train.py \
--dataset=KITTI \
--pretrained_model_path=./data/SqueezeNet/squeezenet_v1.1.pkl \
--data_path=./data/KITTI \
--image_set=train \
--train_dir=/tmp/bichen/logs/SqueezeDet/train \
--net=squeezeDet \
--summary_step=100 \
--checkpoint_step=500 \
--gpu=0

# =========================================================================== #
# command for squeezeDet+:
# =========================================================================== #
# python ./src/train.py \
# --dataset=KITTI \
# --pretrained_model_path=./data/squeezenet/squeezenet_v1.0_SR_0.750.pkl \
# --pretrained_model_path=./data/SqueezeNet/squeezenet_v1.0_SR_0.750.pkl \
# --data_path=./data/KITTI \
# --image_set=train \
# --train_dir=/tmp/bichen/logs/SqueezeDetPlus/train \
Expand All @@ -45,13 +45,13 @@
# =========================================================================== #
# command for resnet50:
# =========================================================================== #
python ./src/train.py \
--dataset=KITTI \
--pretrained_model_path=./data/ResNet/ResNet-50-weights.pkl \

This comment has been minimized.

Sign in to view

Copy link
@gunjansarma

gunjansarma Sep 1, 2023

what was the pre trained model
--data_path=./data/KITTI \
--image_set=train \
--train_dir=/tmp/bichen/logs/resnet/train \
--net=resnet50 \
--summary_step=100 \
--checkpoint_step=500 \
--gpu=0
# python ./src/train.py \
# --dataset=KITTI \
# --pretrained_model_path=./data/ResNet/ResNet-50-weights.pkl \
# --data_path=./data/KITTI \
# --image_set=train \
# --train_dir=/tmp/bichen/logs/resnet/train \
# --net=resnet50 \
# --summary_step=100 \
# --checkpoint_step=500 \
# --gpu=0

0 comments on commit 91ee8b4

Please sign in to comment.