(self)
| 159 | self.parser.add_argument('--path', type=str, default='demo/images/running.png', help='the path of your file') |
| 160 | |
| 161 | def parse(self): |
| 162 | self.init() |
| 163 | |
| 164 | self.opt = self.parser.parse_args() |
| 165 | self.opt.pad = (self.opt.frames - 1) // 2 |
| 166 | |
| 167 | self.opt.subjects_train = "S1,S5,S6,S7,S8" |
| 168 | self.opt.subjects_test = "S9,S11" |
| 169 | |
| 170 | self.opt.root_joint = 0 |
| 171 | if self.opt.dataset == "h36m": |
| 172 | self.opt.subjects_train = "S1,S5,S6,S7,S8" |
| 173 | self.opt.subjects_test = "S9,S11" |
| 174 | |
| 175 | if self.opt.keypoints.startswith("sh") or self.opt.keypoints.startswith( |
| 176 | "hr" |
| 177 | ): |
| 178 | self.opt.n_joints = 16 |
| 179 | self.opt.out_joints = 16 |
| 180 | |
| 181 | self.opt.joints_left = [4, 5, 6, 10, 11, 12] |
| 182 | self.opt.joints_right = [1, 2, 3, 13, 14, 15] |
| 183 | else: |
| 184 | self.opt.n_joints = 17 |
| 185 | self.opt.out_joints = 17 |
| 186 | |
| 187 | self.opt.joints_left = [4, 5, 6, 11, 12, 13] # left side |
| 188 | self.opt.joints_right = [1, 2, 3, 14, 15, 16] |
| 189 | |
| 190 | return self.opt |
no test coverage detected