diff --git a/demo/web_demo.py b/demo/web_demo.py index be81d9b..b56d11e 100644 --- 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 index 464c7ef..8748520 100644 --- 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 = []