KerasObjectDetector
KerasObjectDetector copied to clipboard
[modeling] Add SSD model on dev branch
WIP : Converting a model
Goal
(boxes, scores, labels) = model.predict_detection('sample.jpg')
Working on
Tasks
- [ ]
model = ssd_300()
is for training - [ ] When
model.predict_detection()
is called, theDecodeDetections
layer will be added at the end of themodel
.
To run the test file, the pretrained model weights are needed. Download files from here, and put them in KerasObjectDetector/keras_ssd/pretraind/
folder.
Starting to add SSD model with @hngskj
Use the weights below. PASCAL VOC models: 07+12: SSD300*, SSD512*