2020-11-09 01:57:57 +00:00
|
|
|
import dotenv from "dotenv-extended";
|
2020-12-12 18:44:51 +00:00
|
|
|
import { Trainer } from "../src/trainer";
|
2020-11-08 20:33:07 +00:00
|
|
|
const main = async () => {
|
2020-11-09 01:57:57 +00:00
|
|
|
dotenv.load({
|
|
|
|
silent: false,
|
|
|
|
errorOnMissing: true,
|
|
|
|
});
|
2020-11-27 06:15:46 +00:00
|
|
|
|
2020-12-12 18:44:51 +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
|
|
|
);
|
2020-12-12 18:44:51 +00:00
|
|
|
await trainer.train(true);
|
2020-11-08 20:33:07 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
main();
|