justadudewhohacks / face-api.js

JavaScript API for face detection and face recognition in the browser and nodejs with tensorflow.js
MIT License
16.51k stars 3.68k forks source link

Fetching external videos in browser (fetchImage for <video>) #392

Open pt-br opened 5 years ago

pt-br commented 5 years ago

I've ran into this issue for a couple hours and I ended up editing the dist library adding two new functions called fetchVideo and bufferToVideo that works pretty much like the fetchImage and bufferToImage functions.

I'll leave it here to help somebody else with the same issue and in case someone wants to include it on future releases.

face-api.js

...
exports.fetchVideo = fetchVideo;
...
function fetchVideo(uri) {
        return __awaiter(this, void 0, void 0, function () {
            var res, blob;
            return __generator(this, function (_a) {
                switch (_a.label) {
                    case 0: return [4 /*yield*/, fetchOrThrow(uri)];
                    case 1:
                        res = _a.sent();
                        return [4 /*yield*/, (res).blob()];
                    case 2:
                        blob = _a.sent();
                        if (!blob.type.startsWith('video/')) {
                            throw new Error("fetchVideo - expected blob type to be of type video/*, instead have: " + blob.type + ", for url: " + res.url);
                        }
                        return [2 /*return*/, bufferToVideo(blob)];
                }
            });
        });
    }

function bufferToVideo(buf) {
        return new Promise(function (resolve, reject) {
            if (!(buf instanceof Blob)) {
                return reject('bufferToVideo - expected buf to be of type: Blob');
            }
            var reader = new FileReader();
            reader.onload = function () {
                if (typeof reader.result !== 'string') {
                    return reject('bufferToVideo - expected reader.result to be a string, in onload');
                }
                var video = env.getEnv().createVideoElement();
                video.onloadstart = function () {
                    setTimeout(() => {
                        return resolve(video);
                    }, 100)
                };
                video.onerror = reject;
                video.type = "video/mp4";
                video.autoplay = true;
                video.src = reader.result;
            };
            reader.onerror = reject;
            reader.readAsDataURL(buf);
        });
    }

Usage example:

const videoElement = document.querySelector('video');
    const detections = await faceapi
      .detectAllFaces(await faceapi.fetchVideo(videoElement.src))
      .withFaceLandmarks()
      .withFaceDescriptors();
pt-br commented 5 years ago

@justadudewhohacks Feel free to close this issue if you want (idk if it should stay open or not)

justadudewhohacks commented 5 years ago

Thanks for your solution, if you want to open a PR contributing this method feel free :)

Bea98 commented 4 years ago

is this code applicable to face-api.min.js and if so how can I implement it? Thank you!

rajkishoreandia commented 4 years ago

@pt-br could you please tell me where to place the code properly.if i write it as stand alone methods.i am getting following error in react.Failed to compile. I changed the face-api.js still getting the below error Attempted import error: 'fetchVideo' is not exported from 'face-api.js' (imported as 'faceapi').

msvargas commented 4 years ago

Please add this feature, thanks!

SathishSaminathan commented 4 years ago

is this code applicable to face-api.min.js and if so how can I implement it? Thank you!

have u found the solution bro?

pt-br commented 4 years ago

@Bea98 @rajkishoreandia you guys need to apply this to the dist file directly, not the min. Then you can compress it if you need a .min version.

codepool867 commented 3 years ago

You did do very good work, Thank you!

bettysteger commented 3 years ago

@pt-br how can you get results by frame or second?

Ntambwe1 commented 2 years ago

hello guys, pls can u explain to step by step how to apply and run this code on python app ??, pls guys , i am a new baby in python coding pls guide me, no harsh comments pls.

bettysteger commented 2 years ago

@Ntambwe1 hey, this is a JS plugin, so i don't know if you can run it in python?