Skip to content

Commit e1d0a82

Browse files
committed
update
change some dir path
1 parent 0872a8f commit e1d0a82

File tree

3 files changed

+8
-7
lines changed

3 files changed

+8
-7
lines changed

eval.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -51,12 +51,13 @@
5151
filename = "final-aug.t7"
5252
num_threads = 10
5353

54-
PATH_PREFIX = "/home/yuliang/code/DeepPose-pytorch/results/{}".format(modeltype)
55-
full_name="/home/yuliang/code/DeepPose-pytorch/models/{}/{}".format(modeltype, filename)
54+
PATH_PREFIX = "./results/{}".format(modeltype)
55+
full_name="./models/{}/{}".format(modeltype, filename)
5656
# full_name = "/home/yuliang/code/DeepPose-pytorch/models/demo/mobilenetv2_224x224-robust.t7"
5757
# full_name = "/home/yuliang/code/DeepPose-pytorch/models/demo/resnet18_227x227.t7"
58-
full_name = "/home/yuliang/code/DeepPose-pytorch/models/demo/mobilenet-best.t7"
59-
ROOT_DIR = "/home/yuliang/code/deeppose_tf/datasets/mpii"
58+
# full_name = "/home/yuliang/code/DeepPose-pytorch/models/demo/mobilenet-best.t7"
59+
# ROOT_DIR = "/home/yuliang/code/deeppose_tf/datasets/mpii"
60+
ROOT_DIR = "../deeppose_tf/datasets/mpii"
6061

6162
if modeltype == 'resnet':
6263
input_size = 227

run_webcam.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828

2929
if __name__ == '__main__':
3030

31-
parser = argparse.ArgumentParser(description='deeppose realtime webcam')
31+
parser = argparse.ArgumentParser(description='deeppose realtime webcam.')
3232
parser.add_argument('--model', type=str, default='resnet', help='mobilenet|resnet')
3333
parser.add_argument('--camera', type=int, default=0)
3434

training.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,8 @@
6868
if pretrain:
6969
net = torch.load('./models/%s/%s'%(modeltype,modelname)).cuda(device_id=gpus[0])
7070

71-
ROOT_DIR = "/home/yuliang/code/deeppose_tf/datasets/mpii"
72-
PATH_PREFIX = '/home/yuliang/code/DeepPose-pytorch/models/{}/'.format(modeltype)
71+
ROOT_DIR = "../deeppose_tf/datasets/mpii"
72+
PATH_PREFIX = './models/{}/'.format(modeltype)
7373

7474
train_dataset = PoseDataset(csv_file=os.path.join(ROOT_DIR,'train_joints.csv'),
7575
transform=transforms.Compose([

0 commit comments

Comments
 (0)