search

dividiti / ck-caffe

Collective Knowledge workflow for Caffe to automate installation across diverse platforms and to collaboratively evaluate and optimize Caffe-based workloads across diverse hardware, software and data sets (compilers, libraries, tools, models, inputs):
http://cKnowledge.org
BSD 3-Clause "New" or "Revised" License
193 stars 40 forks source link

SSD support #95

Closed Rhymmor closed 7 years ago

Rhymmor commented 7 years ago

85

In this PR:

  1. Two packages with Caffe SSD libraries (CPU and GPU with CUDA)

    • lib-caffe-ssd-cpu
    • lib-caffe-ssd-cuda

  2. Corresponding programs for SSD detection (only with default mode)

    • caffe-detection-ssd
    • caffe-detection-ssd-cuda

  3. COCO-2014, VOC-2007, VOC-2012 datasets

    • dataset-coco-2014
    • datataset-voc-2007
    • dataset-voc-2012

  4. Pretrained SSD 300 and 512 models (trained on COCO and VOC)

    • caffemodel-ssd-coco-300
    • caffemodel-ssd-coco-512
    • caffemodel-ssd-voc-300
    • caffemodel-ssd-voc-512
dsavenko commented 7 years ago

Will provide an updated PR soon.