search

cetra3 / mtcnn

Tensorflow Rust Implementation of MTCNN - https://cetra3.github.io/blog/face-detection-with-tensorflow-rust/
72 stars 5 forks source link

Image from opencv #5

Closed kalmastenitin closed 1 year ago

kalmastenitin commented 1 year ago

Hi! This is nice implementation. Can we use this use with opencv frame as input instead of image as input? It might helpful in creating pipeline of face detection

cetra3 commented 1 year ago

Yes you should be able to use an opencv frame, but will need to flatten the pixels into BGR data as per what's done here:

https://github.com/cetra3/mtcnn/blob/59783c80b73ec0e3eb2f66f2c2062d5d3ba746c0/src/main.rs#L48-L56