rykov8 / ssd_keras

Port of Single Shot MultiBox Detector to Keras
MIT License
1.1k stars 553 forks source link

SSD for th dim_ordering #126

Closed hkvision closed 6 years ago

hkvision commented 6 years ago

Wondering if there is pretrained weights for th dim_ordering with input_shape=(3, 300, 300)? Also, if I use th, will the code for detection process be different? i.e. function detection_out and the code for visualizing the box and prediction. Thanks so much in advance!