Add training script
Showing
- README.md 24 additions, 1 deletionREADME.md
- data/weights/imagenet-mobilenetv3-large.pth 0 additions, 0 deletionsdata/weights/imagenet-mobilenetv3-large.pth
- data/weights/imagenet-mobilenetv3-small.pth 0 additions, 0 deletionsdata/weights/imagenet-mobilenetv3-small.pth
- synthetic_ssd/config.py 1 addition, 0 deletionssynthetic_ssd/config.py
- synthetic_ssd/datasets/detection_dataset.py 19 additions, 1 deletionsynthetic_ssd/datasets/detection_dataset.py
- synthetic_ssd/models/__init__.py 72 additions, 36 deletionssynthetic_ssd/models/__init__.py
- synthetic_ssd/models/multibox_loss.py 56 additions, 0 deletionssynthetic_ssd/models/multibox_loss.py
- synthetic_ssd/scripts/run_eval.py 3 additions, 1 deletionsynthetic_ssd/scripts/run_eval.py
- synthetic_ssd/scripts/train_ssd.py 260 additions, 0 deletionssynthetic_ssd/scripts/train_ssd.py
- synthetic_ssd/utils/load_weights.py 15 additions, 12 deletionssynthetic_ssd/utils/load_weights.py
Loading
Please register or sign in to comment