homebridge-face-location/scripts/train.ts

17 lines
337 B
TypeScript
Raw Normal View History

import dotenv from "dotenv-extended";
import { Trainer } from "../src/trainer";
2020-11-08 20:33:07 +00:00
const main = async () => {
dotenv.load({
silent: false,
errorOnMissing: true,
});
2020-11-27 06:15:46 +00:00
const trainer = new Trainer(
process.env.REF_IMAGE_DIR as string,
process.env.TRAINED_MODEL_DIR as string
2020-11-27 06:15:46 +00:00
);
await trainer.train(true);
2020-11-08 20:33:07 +00:00
};
main();