| 28 | self.parser = argparse.ArgumentParser() |
| 29 | |
| 30 | def init(self): |
| 31 | self.parser.add_argument("--model", default="", type=str) |
| 32 | self.parser.add_argument("--layers", default=3, type=int) |
| 33 | self.parser.add_argument("--channel", default=512, type=int) |
| 34 | self.parser.add_argument("--d_hid", default=1024, type=int) |
| 35 | self.parser.add_argument("--dataset", type=str, default="h36m") |
| 36 | self.parser.add_argument( |
| 37 | "-k", "--keypoints", default="cpn_ft_h36m_dbb", type=str |
| 38 | ) |
| 39 | self.parser.add_argument("--data_augmentation", type=bool, default=True) |
| 40 | self.parser.add_argument("--reverse_augmentation", type=bool, default=False) |
| 41 | self.parser.add_argument("--test_augmentation", type=str2bool, default=True) |
| 42 | self.parser.add_argument( |
| 43 | "--test_augmentation_flip_hypothesis", type=str2bool, default=False |
| 44 | ) |
| 45 | self.parser.add_argument( |
| 46 | "--test_augmentation_FlowAug", type=str2bool, default=False |
| 47 | ) |
| 48 | self.parser.add_argument("--crop_uv", type=int, default=0) |
| 49 | self.parser.add_argument("--root_path", type=str, default="dataset/") |
| 50 | self.parser.add_argument("-a", "--actions", default="*", type=str) |
| 51 | self.parser.add_argument("--downsample", default=1, type=int) |
| 52 | self.parser.add_argument("--subset", default=1, type=float) |
| 53 | self.parser.add_argument("-s", "--stride", default=1, type=int) |
| 54 | self.parser.add_argument("--gpu", default="0", type=str, help="") |
| 55 | self.parser.add_argument("--train", action="store_true") |
| 56 | self.parser.add_argument("--test", type=int, default=1) # |
| 57 | self.parser.add_argument("--nepoch", type=int, default=41) # |
| 58 | self.parser.add_argument( |
| 59 | "--batch_size", |
| 60 | type=int, |
| 61 | default=128, |
| 62 | help="can be changed depending on your machine", |
| 63 | ) # default 128 |
| 64 | self.parser.add_argument("--lr", type=float, default=1e-3) |
| 65 | self.parser.add_argument("--lr_decay_large", type=float, default=0.5) |
| 66 | self.parser.add_argument("--large_decay_epoch", type=int, default=5) |
| 67 | self.parser.add_argument("--workers", type=int, default=8) |
| 68 | self.parser.add_argument("-lrd", "--lr_decay", default=0.95, type=float) |
| 69 | self.parser.add_argument("--frames", type=int, default=1) # |
| 70 | self.parser.add_argument( |
| 71 | "--pad", type=int, default=175 |
| 72 | ) # pad = (self.opt.frames-1) // 2 |
| 73 | self.parser.add_argument("--reload", action="store_true") |
| 74 | self.parser.add_argument("--model_dir", type=str, default="") |
| 75 | # Optional: load model class from a specific file path |
| 76 | self.parser.add_argument("--model_path", type=str, default="") |
| 77 | # Model registry name (e.g. "fmpose3d_humans"); used instead of --model_path |
| 78 | self.parser.add_argument("--model_type", type=str, default="fmpose3d_humans") |
| 79 | self.parser.add_argument("--model_weights_path", type=str, default="") |
| 80 | |
| 81 | self.parser.add_argument("--post_refine_reload", action="store_true") |
| 82 | self.parser.add_argument("--checkpoint", type=str, default="") |
| 83 | self.parser.add_argument( |
| 84 | "--previous_dir", type=str, default="./pre_trained_model/pretrained" |
| 85 | ) |
| 86 | |
| 87 | self.parser.add_argument("--n_joints", type=int, default=17) |