├── .gitignore ├── .npmignore ├── .travis.yml ├── LICENSE ├── README.md ├── dist ├── tiny-yolov2.js ├── tiny-yolov2.js.map └── tiny-yolov2.min.js ├── examples ├── package-lock.json ├── package.json ├── public │ ├── commons.js │ ├── github_link_icon.png │ ├── images │ │ ├── face_detection │ │ │ ├── 1.jpg │ │ │ ├── 2.jpg │ │ │ ├── 3.jpg │ │ │ ├── 4.jpg │ │ │ └── 5.jpg │ │ └── object_detection │ │ │ ├── 1.jpg │ │ │ ├── 2.jpg │ │ │ ├── 3.jpg │ │ │ ├── 4.jpg │ │ │ ├── 5.jpg │ │ │ └── 6.jpg │ ├── menu_icon.png │ └── styles.css ├── server.js └── views │ ├── cocoObjectDetection.html │ ├── faceDetection.html │ └── vocObjectDetection.html ├── karma.conf.js ├── models ├── coco_model-shard1 ├── coco_model-shard2 ├── coco_model-shard3 ├── coco_model-weights_manifest.json ├── coco_model_config.json ├── face_detection_model-shard1 ├── face_detection_model-weights_manifest.json ├── face_detection_model_config.json ├── voc_model-shard1 ├── voc_model-shard2 ├── voc_model-shard3 ├── voc_model-shard4 ├── voc_model-weights_manifest.json └── voc_model_config.json ├── package.json ├── rollup.config.js ├── src ├── common │ ├── convLayer.ts │ ├── extractConvParamsFactory.ts │ ├── extractFCParamsFactory.ts │ ├── extractSeparableConvParamsFactory.ts │ ├── index.ts │ └── types.ts ├── index.ts └── tinyYolov2 │ ├── TinyYolov2.ts │ ├── TinyYolov2LossFunction.ts │ ├── TinyYolov2Options.ts │ ├── TinyYolov2Trainable.ts │ ├── config.ts │ ├── const.ts │ ├── convWithBatchNorm.ts │ ├── depthwiseSeparableConv.ts │ ├── extractParams.ts │ ├── extractParamsFromWeigthMap.ts │ ├── getDefaultBackwardOptions.ts │ ├── index.ts │ ├── leaky.ts │ └── types.ts ├── test ├── TinyYolov2LossFunction.test.ts └── utils.ts ├── train ├── face │ ├── initWeights.html │ ├── test.html │ ├── train.html │ ├── verify.html │ └── verify_debug.html ├── js │ ├── createLossReporter.js │ └── randomCrop.js ├── package-lock.json ├── package.json ├── serveFaceDetection.js ├── serveVoc.js └── voc │ ├── initWeights.html │ ├── trainVoc.html │ └── verifyVoc.html ├── tsconfig.es6.json ├── tsconfig.json └── tsconfig.test.json /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/.gitignore -------------------------------------------------------------------------------- /.npmignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/.npmignore -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/.travis.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/README.md -------------------------------------------------------------------------------- /dist/tiny-yolov2.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/dist/tiny-yolov2.js -------------------------------------------------------------------------------- /dist/tiny-yolov2.js.map: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/dist/tiny-yolov2.js.map -------------------------------------------------------------------------------- /dist/tiny-yolov2.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/dist/tiny-yolov2.min.js -------------------------------------------------------------------------------- /examples/package-lock.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/package-lock.json -------------------------------------------------------------------------------- /examples/package.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/package.json -------------------------------------------------------------------------------- /examples/public/commons.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/commons.js -------------------------------------------------------------------------------- /examples/public/github_link_icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/github_link_icon.png -------------------------------------------------------------------------------- /examples/public/images/face_detection/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/images/face_detection/1.jpg -------------------------------------------------------------------------------- /examples/public/images/face_detection/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/images/face_detection/2.jpg -------------------------------------------------------------------------------- /examples/public/images/face_detection/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/images/face_detection/3.jpg -------------------------------------------------------------------------------- /examples/public/images/face_detection/4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/images/face_detection/4.jpg -------------------------------------------------------------------------------- /examples/public/images/face_detection/5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/images/face_detection/5.jpg -------------------------------------------------------------------------------- /examples/public/images/object_detection/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/images/object_detection/1.jpg -------------------------------------------------------------------------------- /examples/public/images/object_detection/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/images/object_detection/2.jpg -------------------------------------------------------------------------------- /examples/public/images/object_detection/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/images/object_detection/3.jpg -------------------------------------------------------------------------------- /examples/public/images/object_detection/4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/images/object_detection/4.jpg -------------------------------------------------------------------------------- /examples/public/images/object_detection/5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/images/object_detection/5.jpg -------------------------------------------------------------------------------- /examples/public/images/object_detection/6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/images/object_detection/6.jpg -------------------------------------------------------------------------------- /examples/public/menu_icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/menu_icon.png -------------------------------------------------------------------------------- /examples/public/styles.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/public/styles.css -------------------------------------------------------------------------------- /examples/server.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/server.js -------------------------------------------------------------------------------- /examples/views/cocoObjectDetection.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/views/cocoObjectDetection.html -------------------------------------------------------------------------------- /examples/views/faceDetection.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/views/faceDetection.html -------------------------------------------------------------------------------- /examples/views/vocObjectDetection.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/examples/views/vocObjectDetection.html -------------------------------------------------------------------------------- /karma.conf.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/karma.conf.js -------------------------------------------------------------------------------- /models/coco_model-shard1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/coco_model-shard1 -------------------------------------------------------------------------------- /models/coco_model-shard2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/coco_model-shard2 -------------------------------------------------------------------------------- /models/coco_model-shard3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/coco_model-shard3 -------------------------------------------------------------------------------- /models/coco_model-weights_manifest.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/coco_model-weights_manifest.json -------------------------------------------------------------------------------- /models/coco_model_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/coco_model_config.json -------------------------------------------------------------------------------- /models/face_detection_model-shard1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/face_detection_model-shard1 -------------------------------------------------------------------------------- /models/face_detection_model-weights_manifest.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/face_detection_model-weights_manifest.json -------------------------------------------------------------------------------- /models/face_detection_model_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/face_detection_model_config.json -------------------------------------------------------------------------------- /models/voc_model-shard1: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/voc_model-shard1 -------------------------------------------------------------------------------- /models/voc_model-shard2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/voc_model-shard2 -------------------------------------------------------------------------------- /models/voc_model-shard3: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/voc_model-shard3 -------------------------------------------------------------------------------- /models/voc_model-shard4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/voc_model-shard4 -------------------------------------------------------------------------------- /models/voc_model-weights_manifest.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/voc_model-weights_manifest.json -------------------------------------------------------------------------------- /models/voc_model_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/models/voc_model_config.json -------------------------------------------------------------------------------- /package.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/package.json -------------------------------------------------------------------------------- /rollup.config.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/rollup.config.js -------------------------------------------------------------------------------- /src/common/convLayer.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/common/convLayer.ts -------------------------------------------------------------------------------- /src/common/extractConvParamsFactory.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/common/extractConvParamsFactory.ts -------------------------------------------------------------------------------- /src/common/extractFCParamsFactory.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/common/extractFCParamsFactory.ts -------------------------------------------------------------------------------- /src/common/extractSeparableConvParamsFactory.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/common/extractSeparableConvParamsFactory.ts -------------------------------------------------------------------------------- /src/common/index.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/common/index.ts -------------------------------------------------------------------------------- /src/common/types.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/common/types.ts -------------------------------------------------------------------------------- /src/index.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/index.ts -------------------------------------------------------------------------------- /src/tinyYolov2/TinyYolov2.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/TinyYolov2.ts -------------------------------------------------------------------------------- /src/tinyYolov2/TinyYolov2LossFunction.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/TinyYolov2LossFunction.ts -------------------------------------------------------------------------------- /src/tinyYolov2/TinyYolov2Options.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/TinyYolov2Options.ts -------------------------------------------------------------------------------- /src/tinyYolov2/TinyYolov2Trainable.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/TinyYolov2Trainable.ts -------------------------------------------------------------------------------- /src/tinyYolov2/config.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/config.ts -------------------------------------------------------------------------------- /src/tinyYolov2/const.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/const.ts -------------------------------------------------------------------------------- /src/tinyYolov2/convWithBatchNorm.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/convWithBatchNorm.ts -------------------------------------------------------------------------------- /src/tinyYolov2/depthwiseSeparableConv.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/depthwiseSeparableConv.ts -------------------------------------------------------------------------------- /src/tinyYolov2/extractParams.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/extractParams.ts -------------------------------------------------------------------------------- /src/tinyYolov2/extractParamsFromWeigthMap.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/extractParamsFromWeigthMap.ts -------------------------------------------------------------------------------- /src/tinyYolov2/getDefaultBackwardOptions.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/getDefaultBackwardOptions.ts -------------------------------------------------------------------------------- /src/tinyYolov2/index.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/index.ts -------------------------------------------------------------------------------- /src/tinyYolov2/leaky.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/leaky.ts -------------------------------------------------------------------------------- /src/tinyYolov2/types.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/src/tinyYolov2/types.ts -------------------------------------------------------------------------------- /test/TinyYolov2LossFunction.test.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/test/TinyYolov2LossFunction.test.ts -------------------------------------------------------------------------------- /test/utils.ts: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/test/utils.ts -------------------------------------------------------------------------------- /train/face/initWeights.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/face/initWeights.html -------------------------------------------------------------------------------- /train/face/test.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/face/test.html -------------------------------------------------------------------------------- /train/face/train.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/face/train.html -------------------------------------------------------------------------------- /train/face/verify.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/face/verify.html -------------------------------------------------------------------------------- /train/face/verify_debug.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/face/verify_debug.html -------------------------------------------------------------------------------- /train/js/createLossReporter.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/js/createLossReporter.js -------------------------------------------------------------------------------- /train/js/randomCrop.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/js/randomCrop.js -------------------------------------------------------------------------------- /train/package-lock.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/package-lock.json -------------------------------------------------------------------------------- /train/package.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/package.json -------------------------------------------------------------------------------- /train/serveFaceDetection.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/serveFaceDetection.js -------------------------------------------------------------------------------- /train/serveVoc.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/serveVoc.js -------------------------------------------------------------------------------- /train/voc/initWeights.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/voc/initWeights.html -------------------------------------------------------------------------------- /train/voc/trainVoc.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/voc/trainVoc.html -------------------------------------------------------------------------------- /train/voc/verifyVoc.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/train/voc/verifyVoc.html -------------------------------------------------------------------------------- /tsconfig.es6.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/tsconfig.es6.json -------------------------------------------------------------------------------- /tsconfig.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/tsconfig.json -------------------------------------------------------------------------------- /tsconfig.test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/justadudewhohacks/tfjs-tiny-yolov2/HEAD/tsconfig.test.json --------------------------------------------------------------------------------