Skip to content

JianliZh429/head_box

Repository files navigation

Head Box

Detect tiny heads in image

Installation

$ cd utils && make && make hungarian && cd .. $ python train.py --hypes hypes/lstm_rezoom.json --gpu 0 --logdir output

Restore training

$ python train.py --hypes hypes/lstm_rezoom.json --gpu 0 --logdir output --weights $weights_file

Evaluation

$ python evaluate.py --weights output/overfeat_rezoom_2018_01_02_09.44/save.ckpt-100000 --test_boxes data/brainwash/val_boxes.json

$ python detect.py $data_dir hypes.json $output_dir

Tensorboard

$ tensorboard --logdir output

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published