Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Allow for the model to be stored in the posenet package folder #1

Open
wants to merge 5 commits into
base: feature/behave-like-node
Choose a base branch
from
Open

Allow for the model to be stored in the posenet package folder #1

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
ae089d3
Allow for the model to be stored in the posenet packge folder
keesschollaart81 Jul 21, 2020
7839ce5
Make the converter folder a module
keesschollaart81 Jul 21, 2020
c3f452d
Merge branch 'master' into patch-1
keesschollaart81 Jul 21, 2020
d811ba0
allow user to set the path
keesschollaart81 Jul 22, 2020
31943ba
Update the config as well
keesschollaart81 Jul 22, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Empty file added posenet/converter/__init__.py
View file
Open in desktop
Empty file.
9 changes: 7 additions & 2 deletions posenet/posenet_factory.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,13 +1,12 @@
import tensorflow as tf
import os
import posenet.converter.config as config
import posenet.converter.tfjs2tf as tfjs2tf
from posenet.resnet import ResNet
from posenet.mobilenet import MobileNet
from posenet.posenet import PoseNet


def load_model(model, stride, quant_bytes=4, multiplier=1.0):
def load_model(model, stride, quant_bytes=4, multiplier=1.0, models_root=None):

if model == config.RESNET50_MODEL:
model_cfg = config.bodypix_resnet50_config(stride, quant_bytes)
Expand All @@ -17,8 +16,14 @@ def load_model(model, stride, quant_bytes=4, multiplier=1.0):
print('Loading MobileNet model')

model_path = model_cfg['tf_dir']

if not models_root is None:
model_path = os.path.join(models_root, model_path)
model_cfg['tf_dir'] = model_path

if not os.path.exists(model_path):
print('Cannot find tf model path %s, converting from tfjs...' % model_path)
import posenet.converter.tfjs2tf as tfjs2tf
tfjs2tf.convert(model_cfg)
assert os.path.exists(model_path)

Expand Down