LatticeSemi / Human-Count

Other
0 stars 1 forks source link

Crosslink-NX VGG-SqDet Human-Count

A network backbone with VGG and SqueezeDet detector to Power Human Counting usecase.

Dataset

Note that the dataset should be in KITTI format. Images and labels should be arranged as follows:

Setup Environment

$ pip install -r requirements.txt

Run Training

  1. Configure scripts/train.sh

    • Mainly configure dataset path variable: TRAIN_DATA_DIR
    • Optionally User can change other configurations.
  2. Run command :

    $ ./run

    Freezing Model

    $ python src/genpb.py –ckpt_dir <log directory> --freeze