geopavlakos commited on
Commit
bfd5850
1 Parent(s): b9669b5

Update vitpose_model.py

Browse files
Files changed (1) hide show
  1. vitpose_model.py +1 -0
vitpose_model.py CHANGED
@@ -6,6 +6,7 @@ import numpy as np
6
  import torch
7
  import torch.nn as nn
8
 
 
9
  from mmpose.apis import inference_top_down_pose_model, init_pose_model, process_mmdet_results
10
 
11
  #os.environ["PYOPENGL_PLATFORM"] = "egl"
 
6
  import torch
7
  import torch.nn as nn
8
 
9
+ from mmdet.apis import inference_detector, init_detector
10
  from mmpose.apis import inference_top_down_pose_model, init_pose_model, process_mmdet_results
11
 
12
  #os.environ["PYOPENGL_PLATFORM"] = "egl"