This repo is a LeNet built with tensorflowjs training on mnist dataset. We use dataset provided by google api
yarn install
This package is served by rollup, config is in rollup.config.ts
yarn dev
About LeNet
Use this pad function expand image to 32 * 32 example:
function expandImageTo32(tensor: Tensor4D) {
return pad(tensor, [
[0, 0],
[2, 2],
[2, 2],
[0, 0]
])
}