qtf/__tests__/qtf-posenet.js

command for want to Quick use TensorFlow.js on cli.

const util = require('util');
const rewire = require('rewire')
const tmp = require('tmp');

const exec = util.promisify(require('child_process').exec);
const fsp = require('fs').promises;

const { tf_loader } = require('../src/qtf-tfjs-loader');
const qtf_posenet = require('../src/qtf-posenet.js')
const _qtf_posenet = rewire('../src/qtf-posenet.js')

const test_img = '__tests__/lena.jpg'

beforeAll(async ()=>{
  await tf_loader()
})

describe('posenet by gcp remote model',()=>{
  beforeAll(async ()=>{
    await exec('bash -c "test -d models/posenet && rm -r models/posenet" || :')
  })

  test('load',async ()=>{

    let model = await qtf_posenet.load_model()

    expect(model)
      .toHaveProperty(
        'baseModel.model.modelUrl',
        'https://storage.googleapis.com/tfjs-models/savedmodel/posenet/mobilenet/float/100/model-stride16.json'
      )
  })

  test('save_model',async ()=>{

    await qtf_posenet.save_model()

    let model = await qtf_posenet.load_model()

    expect(model)
      .toHaveProperty(
        'baseModel.model.modelUrl',
        'file://./models/posenet/model.json'
      )
  })
})

describe('posenet by local model',()=>{

  test('run',async ()=>{
    let result = await qtf_posenet.run('__tests__/lena.jpg')

    expect(result)
      .toHaveProperty('keypoints.0.part','nose')

    // maybe value of x is 305.82982733463035
    // othertime 310.44099586575874,
    expect(result.keypoints[0].position.x)
      .toBeGreaterThanOrEqual(300);
    expect(result.keypoints[0].position.x)
      .toBeLessThanOrEqual(315);

    //  maybe value of score is 0.9134889245033264,
    expect(result.keypoints[0].score)
      .toBeGreaterThanOrEqual(0.9);
  })

  test('out_image',async ()=>{
    let result = await qtf_posenet.run(test_img)
    let out_img = tmp.tmpNameSync({ postfix:'.jpg' });

    await expect(fsp.access(out_img)).rejects.toThrow()

    await qtf_posenet.out_image(test_img,out_img,result)

    await expect(fsp.access(out_img)).resolves.toBeUndefined()
  })
})