Public
Edited
Oct 12, 2023
cocoSsd = await import("https://cdn.jsdelivr.net/npm/@tensorflow-models/coco-ssd@2.2.3/+esm")
tf = await import("https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@4.11.0/+esm")
loadModel = async () => {
const model = await cocoSsd.load();
return model;
};
modifyModel = (model) => {
const newModel = tf.model({
inputs: model.inputs,
outputs: model.layers[model.layers.length - 2].output
});
return newModel;
};
compileModel = (model) => {
model.compile({
optimizer: tf.train.adam(),
loss: 'categoricalCrossentropy',
metrics: ['accuracy']
});
};
trainModel = async (model, trainData, valData) => {
await model.fit(trainData, {
epochs: 10,
validationData: valData
});
};
run = async () => {
const model = await loadModel();
const modifiedModel = modifyModel(model);
compileModel(modifiedModel);
// Assume trainData and valData are prepared beforehand
// await trainModel(modifiedModel, trainData, valData);
};
model = await loadModel();
Purpose-built for displays of data
Observable is your go-to platform for exploring data and creating expressive data visualizations. Use reactive JavaScript notebooks for prototyping and a collaborative canvas for visual data exploration and dashboard creation.
