Giter VIP home page Giter VIP logo

insightface-kotlin's Introduction

简介

尝试将insightface改为kotlin实现,使用了ND4J连实行numpy的操作。

同时用javacp来完成其中opencv的操作。对于skimage中的相似性变换,用javacp中的estimateAffinePartial2D来替代。

模型数据使用insightface的训练模型,由于都放到一个目录下了,所以model-r100-ii的模型名改为face_model,gamodel-r50的改为ga_model。 mtcnn的模型继续使用det1~4的名称。

默认假设

有以下几个改动,

  1. 默认num_worker为1,所以代码上没有大于1的对应
  2. 假设det type永远为0
  3. 最后,mtcnn阶段的extended stage还没有写(默认accurate landmark为false)

问题

目前加载mtcnn的det1~4(即PNet,RNet等),然后执行到PNet的predict会报错(std:bad_alloc)。已与Python进行了一对一的debug, 传入参数一模一样,python正常,但是scala接口就会报该错。

重现方法,执行UnitTest中的FaceModeTest.getFaceModel()

insightface-kotlin's People

Contributors

mikeshi80 avatar

Watchers

James Cloos avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.