oracle / graphpipe

Machine Learning Model Deployment Made Simple
https://oracle.github.io/graphpipe
Other
720 stars 103 forks source link

ssd yolo #13

Open chl916185 opened 5 years ago

chl916185 commented 5 years ago

Can deployments use the yolo SSD faster- RCNN model ? @termie @sleepsonthefloor @Djelibeybi @vishvananda @tjfontaine

sleepsonthefloor commented 5 years ago

There are some pretrained object detection models here:

https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md

I previously wrote some instructions on getting ssd_mobilenet to work here: https://github.com/oracle/graphpipe/issues/9

You may have to adjust input/output names for your chosen model.