circe 0.1.0

data/ext/yunet.h

@@ -0,0 +1,38 @@
+#ifndef __YUNET__
+#define __YUNET__
+
+#include <string>
+#include <vector>
+#include <array>
+#include <utility>
+
+#include <opencv2/dnn.hpp>
+
+
+class YuNet
+{
+
+public:
+    typedef std::array<cv::Point, 5> Landmark;
+    typedef std::pair<cv::Rect, Landmark> Face;
+    
+private:
+    static constexpr int                MODEL_WIDTH    = 512;
+    static constexpr float              CONF_THRESHOLD = 0.4f;
+    static constexpr float              NMS_THRESHOLD  = 0.3f;
+
+    const std::vector<float>            VARIANCES      = { 0.1f, 0.2f };
+    const std::vector<int>              STEPS          = { 8, 16, 32, 64 };
+    const std::vector<std::vector<int>> MIN_SIZES      = {
+	{ 10, 16, 24 }, { 32, 48 }, { 64, 96 }, { 128, 192, 256 } };
+
+public:
+    YuNet(const std::string& model);
+    ~YuNet() {};
+    void process(const cv::Mat& img, std::vector<Face>& faces);
+
+private:
+    cv::dnn::Net net;
+};
+
+#endif

data/lib/circe/version.rb

@@ -0,0 +1,3 @@
+class Circe
+    VERSION = '0.1.0'
+end

data/lib/circe.rb

@@ -0,0 +1,13 @@
+class Circe
+
+    private
+    
+    # Don't know how to do it inside the c extension
+    DATA_DIR   = File.join(__dir__, '..', 'data').freeze
+    ONNX_YOLO  = File.join(DATA_DIR, 'yolov5s.onnx')
+    ONNX_YUNET = File.join(DATA_DIR, 'face_detection_yunet_2022mar.onnx')
+    
+end
+
+require_relative 'circe/version'
+require_relative 'circe/core'

metadata

@@ -0,0 +1,60 @@
+--- !ruby/object:Gem::Specification
+name: circe
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Stéphane D'Alu
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2023-02-03 00:00:00.000000000 Z
+dependencies: []
+description: |2+
+
+  Face and object recognition
+
+email:
+- sdalu@sdalu.com
+executables: []
+extensions:
+- ext/extconf.rb
+extra_rdoc_files: []
+files:
+- circe.gemspec
+- data/face_detection_yunet_2022mar.onnx
+- data/yolov5s.onnx
+- ext/camera_model.h
+- ext/circe.cpp
+- ext/extconf.rb
+- ext/yolo.cpp
+- ext/yolo.h
+- ext/yunet.cpp
+- ext/yunet.h
+- lib/circe.rb
+- lib/circe/version.rb
+homepage: https://github.com/sdalu/ruby-circe
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.2
+signing_key:
+specification_version: 4
+summary: Face and object recognition
+test_files: []
+...