-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.html
66 lines (51 loc) · 1.9 KB
/
index.html
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
<html>
<body>
<!-- Se carga el JS de la class ONNX -->
<script src="https://cdn.jsdelivr.net/npm/onnxjs/dist/onnx.min.js"></script>
<!-- script src="onnx.min.js"></script -->
<script>
// Clase Modelo utilizada para cargar modelo ONNX y ejecutar
class Modelo{
constructor() {
this.sess = new onnx.InferenceSession()
}
async load(modelo) {
await this.sess.loadModel(modelo)
this.inputName = this.sess.session._model._graph._allInputNames
this.inputType = this.sess.session._model._graph._allData[0].type.tensorType
this.inputDim = this.sess.session._model._graph._allData[0].type.shape.dims
console.log('inputName:', this.inputName)
console.log('inputType:', this.inputType)
console.log('inputDim:', this.inputDim)
}
async run(datos) {
//datos = [1.33]
// let input = new onnx.Tensor(new Float32Array(datos), 'float32', [1,1])
let input = new onnx.Tensor(new Float32Array(datos), this.inputType, this.inputDim)
this.inputTensor = input
this.outputMap = await this.sess.run([input])
this.outputTensor = this.outputMap.values().next().value
console.log('Tensor salida:', this.outputTensor.data[0])
}
}
// Se crea el objeto modelo
modelo = new Modelo();
// Se carga el modelo desde archivo onnx
file='./custominverse.onnx.jpg';
modelo.load(file);
// Espera que el modelo este cargado para seguir
console.log('modelo_cargado:', modelo.sess.session._initialized);
// Se ejecuta el modelo con los parametros de entrada solo si ya se ha inicializado
setTimeout(()=>{ // espera un segundo
if(modelo.sess.session._initialized){
datos = [1.2];
modelo.run(datos);
// Se imprimen los resultados de prediccion y los parametros utilizados
console.log('inputTensor:', modelo.inputTensor);
console.log('outputTensor:', modelo.outputTensor);
console.log('outMap:', modelo.outputMap);
}
}, 1000); // espera un segundo
</script>
</body>
</html>