Skip to content

Commit 89ff316

Browse files
committed
Refactor vis_in_the_wild.py: remove redundant import of resolve_weights_path and add it back in a more appropriate location for better code organization.
1 parent 52c2497 commit 89ff316

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

demo/vis_in_the_wild.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from fmpose3d.lib.checkpoint.download_checkpoints import ensure_checkpoints
2020
ensure_checkpoints()
2121

22+
from fmpose3d.utils.weights import resolve_weights_path
2223
from fmpose3d.lib.preprocess import h36m_coco_format, revise_kpts
2324
from fmpose3d.lib.hrnet.gen_kpts import gen_video_kpts as hrnet_pose
2425
from fmpose3d.common.arguments import opts as parse_args
@@ -278,7 +279,6 @@ def get_pose3D(path, output_dir, type='image'):
278279

279280
# if args.reload:
280281
model_dict = model['CFM'].state_dict()
281-
from fmpose3d.utils.weights import resolve_weights_path
282282
model_path = resolve_weights_path(args.model_weights_path, args.model_type)
283283

284284
print(f"Loading weights from: {model_path}")

0 commit comments

Comments
 (0)