KerasKorea / KerasObjectDetector

Keras Object Detection API with YOLK project 🍳

MIT License

112 stars 48 forks source link

[modeling] Add SSD model on dev branch #31

Open hngskj opened 5 years ago

hngskj commented 5 years ago

Add SSD model on dev branch

Github repo: ssd_keras by @pierluigiferrari

hngskj commented 5 years ago

WIP : Converting a model

Goal

(boxes, scores, labels) = model.predict_detection('sample.jpg')

Working on

test_load_weights.py
keras_ssd300.py
ssd_detection.py
keras_layer_DecodeDetections.py

Tasks

[ ] model = ssd_300() is for training
[ ] When model.predict_detection() is called, the DecodeDetections layer will be added at the end of the model.

hngskj commented 5 years ago

To run the test file, the pretrained model weights are needed. Download files from here, and put them in KerasObjectDetector/keras_ssd/pretraind/ folder.

parkjh688 commented 5 years ago

Starting to add SSD model with @hngskj

parkjh688 commented 5 years ago

Use the weights below. PASCAL VOC models: 07+12: SSD300, SSD512