Skip to content

Commit

Permalink
--
Browse files Browse the repository at this point in the history
  • Loading branch information
rainLiuplus committed Sep 14, 2018
1 parent 5ade339 commit c4cf3d1
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions mmdnn/conversion/examples/tensorflow/extractor.py
Expand Up @@ -165,15 +165,15 @@ class tensorflow_extractor(base_extractor):
'input' : lambda : tf.placeholder(name='input', dtype=tf.float32, shape=[None, 331, 331, 3]),
'num_classes' : 1001,
},
'facenet' : {
'url' : 'https://drive.google.com/facenet_frozen',
'filename' : 'facenet.pb',
'tensor_out' : 'embeddings:0',
'tensor_in' : 'input:0',
'phase_train' : 'phase_train:0',
'input_shape' : [160, 160, 3],
'num_classes' : 0,
},
# 'facenet' : {
# 'url' : 'https://drive.google.com/facenet_frozen',
# 'filename' : 'facenet.pb',
# 'tensor_out' : 'embeddings:0',
# 'tensor_in' : 'input:0',
# 'phase_train' : 'phase_train:0',
# 'input_shape' : [160, 160, 3],
# 'num_classes' : 0,
# },
# Note that the link will expire after one day and it is better to upload the model to a stable server.
'facenet_frozen' : {
'url' : 'https://doc-0k-7k-docs.googleusercontent.com/docs/securesc/ha0ro937gcuc7l7deffksulhg5h7mbp1/bilqocet7fepe2lnqpoacs1gkhrv2kq8/1536890400000/18056234690049221457/*/1R77HmFADxe87GmoLwzfgMu_HY0IhcyBz?e=download',
Expand Down

0 comments on commit c4cf3d1

Please sign in to comment.