Skip to content

Commit a97cd22

Browse files
justadudewhohacksjustadudewhohacks
committed
qunatized tiny_yolov2_separable_conv_model
1 parent 3f4aa56 commit a97cd22

File tree

4 files changed

+10
-6
lines changed

4 files changed

+10
-6
lines changed

‎tools/quantize/server.js

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,10 @@ app.use(express.static(path.join(__dirname, './public')))
99
app.use(express.static(path.join(__dirname, './node_modules/file-saver')))
1010
app.use(express.static(path.join(__dirname, '../../examples/public')))
1111
app.use(express.static(path.join(__dirname, '../../weights')))
12+
app.use(express.static(path.join(__dirname, '../../weights_uncompressed')))
1213
app.use(express.static(path.join(__dirname, '../../dist')))
1314

14-
app.get('/', (req, res) => res.redirect('/quantize_tiny_yolov2'))
15-
app.get('/quantize_tiny_yolov2', (req, res) => res.sendFile(path.join(viewsDir, 'quantizeTinyYolov2.html')))
15+
app.get('/', (req, res) => res.redirect('/quantize_model'))
16+
app.get('/quantize_model', (req, res) => res.sendFile(path.join(viewsDir, 'quantizeModel.html')))
1617

1718
app.listen(3000, () => console.log('Listening on port 3000!'))

‎tools/quantize/views/quantizeTinyYolov2.html renamed to ‎tools/quantize/views/quantizeModel.html

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,10 @@
1010
<script>
1111
tf = faceapi.tf
1212

13+
const modelName = 'tiny_yolov2_separable_conv'
14+
const uncompressedWeightsUri = `tiny_yolov2_separable_conv_model_v1.weights`
15+
const net = new faceapi.TinyYolov2(true)
16+
1317
async function loadNetWeights(uri) {
1418
return new Float32Array(await (await fetch(uri)).arrayBuffer())
1519
}
@@ -36,15 +40,13 @@
3640
}
3741

3842
async function quantizeAndSave() {
39-
const modelName = 'tiny_yolov2'
4043

41-
const tinyYolov2 = new faceapi.TinyYolov2()
42-
await tinyYolov2.load(await loadNetWeights(`uncompressed/tiny_yolov2_model.weights`))
44+
await net.load(await loadNetWeights(uncompressedWeightsUri))
4345

4446
const quantizedTensorArrays = []
4547
const weightEntries = []
4648

47-
tinyYolov2.getParamList().forEach(({ path, tensor }) => {
49+
net.getParamList().forEach(({ path, tensor }) => {
4850
const { scale, min, qdata } = quantizeWeights(tensor)
4951

5052
const weightEntry = {

‎weights/tiny_yolov2_separable_conv_model-shard1

1.71 MB
Binary file not shown.

‎weights/tiny_yolov2_separable_conv_model-weights_manifest.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
[{"weights":[{"name":"conv0/depthwise_filter","shape":[3,3,3,1],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.004699238725737029,"min":-0.7471789573921876}},{"name":"conv0/pointwise_filter","shape":[1,1,3,16],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.008118405529097015,"min":-1.071629529840806}},{"name":"conv0/bias","shape":[16],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.0024678509609372006,"min":-0.28873856242965246}},{"name":"conv1/depthwise_filter","shape":[3,3,16,1],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.004553892331964829,"min":-0.5737904338275684}},{"name":"conv1/pointwise_filter","shape":[1,1,16,32],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.00980057996862075,"min":-1.3230782957638012}},{"name":"conv1/bias","shape":[32],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.0011220066278588537,"min":-0.20644921952602907}},{"name":"conv2/depthwise_filter","shape":[3,3,32,1],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.0032098570290733787,"min":-0.38839270051787883}},{"name":"conv2/pointwise_filter","shape":[1,1,32,64],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.008682825051101984,"min":-1.154815731796564}},{"name":"conv2/bias","shape":[64],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.0015120926440930834,"min":-0.21471715546121783}},{"name":"conv3/depthwise_filter","shape":[3,3,64,1],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.003597520496331009,"min":-0.4317024595597211}},{"name":"conv3/pointwise_filter","shape":[1,1,64,128],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.010341314240997913,"min":-1.3650534798117246}},{"name":"conv3/bias","shape":[128],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.002109630785736383,"min":-0.4113780032185947}},{"name":"conv4/depthwise_filter","shape":[3,3,128,1],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.004783747476689955,"min":-0.6171034244930043}},{"name":"conv4/pointwise_filter","shape":[1,1,128,256],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.009566552498761345,"min":-1.2627849298364977}},{"name":"conv4/bias","shape":[256],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.0020002245903015135,"min":-0.3860433459281921}},{"name":"conv5/depthwise_filter","shape":[3,3,256,1],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.004355777244941861,"min":-0.4791354969436047}},{"name":"conv5/pointwise_filter","shape":[1,1,256,512],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.010036561068366555,"min":-1.2545701335458193}},{"name":"conv5/bias","shape":[512],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.0023248311935686597,"min":-0.42776893961663337}},{"name":"conv6/depthwise_filter","shape":[3,3,512,1],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.004659063442080629,"min":-0.5963601205863205}},{"name":"conv6/pointwise_filter","shape":[1,1,512,1024],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.010061494509379069,"min":-1.2576868136723836}},{"name":"conv6/bias","shape":[1024],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.0029680932269376867,"min":-0.3947563991827123}},{"name":"conv7/depthwise_filter","shape":[3,3,1024,1],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.003887363508635876,"min":-0.48980780208812036}},{"name":"conv7/pointwise_filter","shape":[1,1,1024,1024],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.009973861189449535,"min":-1.2766542322495404}},{"name":"conv7/bias","shape":[1024],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.004667898486642276,"min":-0.6955168745096991}},{"name":"conv8/filters","shape":[1,1,1024,25],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.01573958116419175,"min":-2.5340725674348716}},{"name":"conv8/bias","shape":[25],"dtype":"float32","quantization":{"dtype":"uint8","scale":0.009396760662396749,"min":-2.2552225589752197}}],"paths":["tiny_yolov2_separable_conv_model-shard1"]}]

0 commit comments

Comments
 (0)