(model, model_weights)
| 59 | |
| 60 | |
| 61 | def transfer(model, model_weights): |
| 62 | transfered_model_weights = {} |
| 63 | for weights_name in model.state_dict().keys(): |
| 64 | transfered_model_weights[weights_name] = model_weights['.'.join(weights_name.split('.')[1:])] |
| 65 | return transfered_model_weights |
| 66 | |
| 67 | |
| 68 | def draw_bodypose(canvas, candidate, subset): |