KerasObjectDetector icon indicating copy to clipboard operation
KerasObjectDetector copied to clipboard

[modeling] Add SSD model on dev branch

Open hngskj opened this issue 4 years ago • 4 comments

Add SSD model on dev branch

Github repo: ssd_keras by @pierluigiferrari

hngskj avatar Oct 12 '19 12:10 hngskj

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, the DecodeDetections layer will be added at the end of the model.

hngskj avatar Oct 16 '19 06:10 hngskj

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

hngskj avatar Oct 16 '19 06:10 hngskj

Starting to add SSD model with @hngskj

parkjh688 avatar Oct 18 '19 05:10 parkjh688

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

parkjh688 avatar Oct 18 '19 06:10 parkjh688