Movenet refactor

example/movenet/Dockerfile

@@ -92,7 +92,7 @@ RUN apt install -y \
   cmake \
   && apt-get autoremove \
   && apt-get clean \
-  && rm -rf /tmp/* /var/lib/apt/lists/* /var/tmp/*
+  && rm -rf /tmp/* /var/moveenet/apt/lists/* /var/tmp/*
 
 RUN pip3 install \
   scikit-build

example/movenet/config.py

@@ -55,6 +55,7 @@
     "eval_outputs": home + '/dhp-outputs/',
     "eval_label_path": home + "/dhp19_eros/poses.json",
     'train_label_path': '',
-    'val_label_path': ''
+    'val_label_path': '',
+    'ckpt': ''
 }
 

example/movenet/evaluate.py

@@ -7,10 +7,11 @@
 import sys
 
 sys.path.append('.')
-from lib import init, Data, MoveNet, Task
+sys.path.append('../../../hpe-core')
+from pycore.moveenet import init, Data, MoveNet, Task
 
 from config import cfg
-from lib.utils.utils import arg_parser
+from pycore.moveenet.utils.utils import arg_parser
 
 
 def main(cfg):

example/movenet/movenet_online.py

@@ -1,8 +1,8 @@
-from lib import init, MoveNet, Task
+from pycore.moveenet import init, MoveNet, Task
 
 from config import cfg
-from lib.utils.utils import arg_parser
-from lib.task.task_tools import image_show, write_output, superimpose
+from pycore.moveenet.utils.utils import arg_parser
+from pycore.moveenet.task.task_tools import image_show, write_output, superimpose
 
 import sys
 import yarp

example/movenet/predict.py

@@ -3,10 +3,10 @@
 https://github.com/fire717
 """
 
-from lib import init, Data, MoveNet, Task
+from pycore.moveenet import init, Data, MoveNet, Task
 
 from config import cfg
-from lib.utils.utils import arg_parser
+from pycore.moveenet.utils.utils import arg_parser
 
 
 # Script to create and save as images all the various outputs of the model

pycore/moveenet/__init__.py

@@ -0,0 +1,30 @@
+"""
+@Fire
+https://github.com/fire717
+"""
+import os
+
+from pycore.moveenet.data.data import Data
+from pycore.moveenet.models.movenet_mobilenetv2 import MoveNet
+from pycore.moveenet.task.task import Task
+
+
+from pycore.moveenet.utils.utils import setRandomSeed, printDash
+
+
+def init(cfg):
+
+    if cfg["cfg_verbose"]:
+        printDash()
+        print(cfg)
+        printDash()
+
+
+
+    os.environ["CUDA_VISIBLE_DEVICES"] = cfg['GPU_ID']
+    setRandomSeed(cfg['random_seed'])
+
+
+    #
+    # if not os.path.exists(cfg['save_dir']):
+    #     os.makedirs(cfg['save_dir'])

pycore/moveenet/config.py

@@ -0,0 +1,59 @@
+"""
+@Fire
+https://github.com/fire717
+"""
+
+cfg = {
+    ##### Global Setting
+    'GPU_ID': '0,1',
+    "num_workers": 4,
+    "random_seed": 42,
+    "cfg_verbose": True,
+    "save_dir": '',
+    "num_classes": 13,
+    "width_mult": 1.0,
+    "img_size": 192,
+    'label': '',
+
+    ##### Train Setting
+    'pre-separated_data': True,
+    'training_data_split': 80,
+    "dataset": '',
+    'balance_data': False,
+    'log_interval': 10,
+    'save_best_only': False,
+
+    'pin_memory': True,
+    'newest_ckpt': '',
+    'th': 50,  # percentage of headsize
+    'from_scratch': True,
+
+    ##### Train Hyperparameters
+    'learning_rate': 0.001,  # 1.25e-4
+    'batch_size': 64,
+    'epochs': 300,
+    'optimizer': 'Adam',  # Adam  SGD
+    'scheduler': 'MultiStepLR-70,100-0.1',  # default  SGDR-5-2  CVPR   step-4-0.8 MultiStepLR
+    # multistepLR-<<>milestones>-<<decay multiplier>>
+    'weight_decay': 0.001,  # 5.e-4,  # 0.0001,
+    'class_weight': None,  # [1., 1., 1., 1., 1., 1., 1., ]
+    'clip_gradient': 5,  # 1,
+    'w_heatmap': 1,
+    'w_bone': 20,
+    'w_center': 1,
+    'w_reg': 3,
+    'w_offset': 1,
+
+    ##### File paths
+    'predict_output_path': '',
+    'results_path': '',
+    "img_path": '',
+    "test_img_path": '',
+    "eval_img_path": '',
+    "eval_outputs": '',
+    "eval_label_path": '',
+    'train_label_path': '',
+    'val_label_path': '',
+    'ckpt': ''
+}
+

example/movenet/lib/data/data.py → pycore/moveenet/data/data.py

@@ -8,8 +8,8 @@
 import json
 import copy
 
-from lib.data.data_tools import getDataLoader, getFileNames
-from lib.task.task_tools import movenetDecode
+from pycore.moveenet.data.data_tools import getDataLoader, getFileNames
+from pycore.moveenet.task.task_tools import movenetDecode
 
 
 class Data():