Skip to content

Commit

Permalink
more DNN models, PyDetectionDNN update for OpenCV 4.5.4
Browse files Browse the repository at this point in the history
  • Loading branch information
jevois committed May 12, 2022
1 parent 2fb3614 commit 0342dd3
Show file tree
Hide file tree
Showing 5 changed files with 259 additions and 116 deletions.
2 changes: 1 addition & 1 deletion CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ include_directories("include")
include_directories("..")

# Setup our library using the source files in src/Components and call it jevoisbase with the current version number:
set(JEVOISBASE_SOVERSION "1.16.0")
set(JEVOISBASE_SOVERSION "1.17.0")
jevois_setup_library(src/Components ${JEVOIS}base ${JEVOISBASE_SOVERSION})

# Check that we are not trying to compile a newer jevoisbase against an older jevois-sdk, which would typically fail:
Expand Down
2 changes: 1 addition & 1 deletion doc/doxygen.cfg
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ PROJECT_NAME = "JeVoisBase"
# could be handy for archiving the generated documentation or if some version
# control system is used.

PROJECT_NUMBER = 1.16
PROJECT_NUMBER = 1.17

# Using the PROJECT_BRIEF tag one can provide an optional one line description
# for a project that appears at the top of each page and should give viewer a
Expand Down
59 changes: 31 additions & 28 deletions share-jevois/dnn/models.yml
Original file line number Diff line number Diff line change
Expand Up @@ -65,14 +65,15 @@ SqueezeNet:
classoffset: 1

# Inception V3 on CPU:
Inception-V3:
preproc: Blob
nettype: OpenCV
postproc: Classify
model: "opencv-dnn/classification/inception_v3_2016_08_28_frozen.pb"
intensors: "NCHW:32F:1x3x299x299"
rgb: false
classes: "opencv-dnn/classification/imagenet_slim_labels.txt"
# too big for JeVois-A33, runs out of memory...
#Inception-V3:
# preproc: Blob
# nettype: OpenCV
# postproc: Classify
# model: "opencv-dnn/classification/inception_v3_2016_08_28_frozen.pb"
# intensors: "NCHW:32F:1x3x299x299"
# rgb: false
# classes: "opencv-dnn/classification/imagenet_slim_labels.txt"

# Googlenet from https://github.com/BVLC/caffe/tree/master/models/bvlc_googlenet on CPU
GoogleNet:
Expand Down Expand Up @@ -134,18 +135,19 @@ OpenCV-Face:
scorescale: "1.0"
classes: "opencv-dnn/detection/face.txt"

# Too big/slow for jevois-a33
# OpenCV's YOLOv3 on CPU
YoloV3-CPU:
preproc: Blob
nettype: OpenCV
postproc: Detect
detecttype: YOLO
model: "darknet/yolo/weights/yolov3.weights"
config: "darknet/yolo/cfg/yolov3.cfg"
intensors: "NCHW:32F:1x3x416x416"
mean: "0 0 0"
scale: 0.00392
classes: "darknet/yolo/data/coco.names"
#YoloV3-CPU:
# preproc: Blob
# nettype: OpenCV
# postproc: Detect
# detecttype: YOLO
# model: "darknet/yolo/weights/yolov3.weights"
# config: "darknet/yolo/cfg/yolov3.cfg"
# intensors: "NCHW:32F:1x3x416x416"
# mean: "0 0 0"
# scale: 0.00392
# classes: "darknet/yolo/data/coco.names"

# OpenCV's MobileNet-SSD Caffe on CPU
SSD-Caffe:
Expand Down Expand Up @@ -179,13 +181,14 @@ SSD-Caffe:
# scale: 0.00392
# classes: "opencv-dnn/segmentation/enet-classes.txt"

DeepLabV3-CPU:
preproc: Blob
nettype: OpenCV
postproc: Segment
segtype: Classes2
model: "opencv-dnn/segmentation/opt_deeplabv3_mnv2_513.pb"
intensors: "NCHW:32F:1x3x513x513"
mean: "127.5 127.5 127.5"
scale: 0.007843
# too slow for jevois-a33, 6.5s/inference...
#DeepLabV3-CPU:
# preproc: Blob
# nettype: OpenCV
# postproc: Segment
# segtype: Classes2
# model: "opencv-dnn/segmentation/opt_deeplabv3_mnv2_513.pb"
# intensors: "NCHW:32F:1x3x513x513"
# mean: "127.5 127.5 127.5"
# scale: 0.007843

Loading

0 comments on commit 0342dd3

Please sign in to comment.