tkwoo / keras_detection

Keras Detection

Really simple detection

Installation

  pip install fvcore
  pip install tensorflow-gpu # > 2.1
  pip install autokeras
  pip install tensorflow-model-optimization

Configuration

All possible options are in detection/configs/defaults.py

# In configs/baseline.yaml
  MODEL:
    NAME: 'resnet50'
  TRAIN_DIR: ('data/train', 'data/train_0513')
  BATCH_SIZE: 48
  OUTPUT_DIR: 'baseline'

Train

# baseline training
python train.py --config-file configs/baseline.yaml

# mixed precision training
python train.py --config-file configs/mixed_precision.yaml

# autokeras training
# not supported currently because of generator issue
# python train.py --config-file configs/baseline_auto.yaml

# OOD training
python train.py --config-file configs/baseline_unk.yaml

Export

Find best model in the saved directory and export.

python export.py --config-file configs/baseline.yaml