fix webdemo

tensorboy · Aug 24, 2018 · a059294 · a059294
1 parent 452edd5
commit a059294
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/demo/web_demo.py b/demo/web_demo.py
@@ -22,7 +22,7 @@
                                               rtpose_preprocess,
                                               ssd_preprocess, vgg_preprocess)
 from network import im_transform
-from evaluation.coco_eval import get_multiplier, get_outputs
+from evaluate.coco_eval import get_multiplier, get_outputs
 #parser = argparse.ArgumentParser()
 #parser.add_argument('--t7_file', required=True)
 #parser.add_argument('--pth_file', required=True)

diff --git a/train_VGG19.py b/train_VGG19.py
@@ -72,6 +72,7 @@
 
 params_transform['np'] = 56
 params_transform['sigma'] = 7.0
+params_transform['limb_width'] = 1.
 
 def build_names():
     names = []