wbenbihi / hourglasstensorflow

Tensorflow implementation of Stacked Hourglass Networks for Human Pose Estimation
MIT License
480 stars 179 forks source link

KeyError: 'nfeats' #29

Closed monajalal closed 2 years ago

monajalal commented 6 years ago

I am confused. How do you get the visualization of estimated pose using pretrained models on an image?

[jalal@goku hourglasstensorlfow]$ ls -ltra
total 1048044
-rw-r--r--.  1 jalal cs-grad 375890392 Apr  4  2017 YOLO_small.ckpt
-rw-r--r--.  1 jalal cs-grad     68410 Aug 11  2017 hg_refined_200.index
-rw-r--r--.  1 jalal cs-grad 173234692 Aug 11  2017 hg_refined_200.data-00000-of-00001
drwxr-xr-x. 15 jalal cs-grad      4096 Mar 25 04:19 ..
-rw-r--r--.  1 jalal cs-grad     20762 Mar 25 04:19 datagen.py
-rw-r--r--.  1 jalal cs-grad      1131 Mar 25 04:19 config.py
-rw-r--r--.  1 jalal cs-grad       817 Mar 25 04:19 config.cfg
-rw-r--r--.  1 jalal cs-grad      5835 Mar 25 04:19 README.txt
-rw-r--r--.  1 jalal cs-grad      6646 Mar 25 04:19 README.md
-rw-r--r--.  1 jalal cs-grad      4573 Mar 25 04:19 filters.py
-rw-r--r--.  1 jalal cs-grad   6055266 Mar 25 04:19 dataset.txt
-rw-r--r--.  1 jalal cs-grad      9257 Mar 25 04:19 inference.py
-rw-r--r--.  1 jalal cs-grad     35449 Mar 25 04:19 hourglass_tiny.py
-rw-r--r--.  1 jalal cs-grad   9161356 Mar 25 04:19 opencv_ffmpeg.dll
-rw-r--r--.  1 jalal cs-grad   9219038 Mar 25 04:19 opencv_ffmpeg_64.dll
-rw-r--r--.  1 jalal cs-grad     46899 Mar 25 04:19 p4frames.mat
-rw-r--r--.  1 jalal cs-grad      1130 Mar 25 04:19 timer.py
-rw-r--r--.  1 jalal cs-grad     51242 Mar 25 04:19 predictClass.py
-rw-r--r--.  1 jalal cs-grad    170640 Mar 25 04:19 pckhfULL.png
-rw-r--r--.  1 jalal cs-grad     11471 Mar 25 04:19 yolo_net.py
drwxr-xr-x.  2 jalal cs-grad       145 Mar 25 04:19 v0.1
-rw-r--r--.  1 jalal cs-grad      2041 Mar 25 04:19 train_launcher.py
drwxr-xr-x.  8 jalal cs-grad       211 Mar 25 04:19 .git
-rw-r--r--.  1 jalal cs-grad 420411266 Mar 25 04:20 DHPE-RegularHG-01283493-Walid-BENBIHI.zip
drwxr-xr-x.  2 jalal cs-grad        54 Mar 25 04:21 DHPE-RegularHG-01283493-Walid-BENBIHI
drwxr-xr-x.  6 jalal cs-grad      4096 Mar 25 04:22 .
drwxr-xr-x.  2 jalal cs-grad       235 Mar 25 04:22 __pycache__
[jalal@goku hourglasstensorlfow]$ /scratch/sjn/anaconda/bin/python  inference.py 
/scratch/sjn/anaconda/lib/python3.6/site-packages/h5py/__init__.py:36: FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
  from ._conv import register_converters as _register_converters
[jalal@goku hourglasstensorlfow]$ ls
DHPE-RegularHG-01283493-Walid-BENBIHI      README.txt       config.cfg  dataset.txt                         hg_refined_200.index  opencv_ffmpeg.dll     pckhfULL.png     train_launcher.py
DHPE-RegularHG-01283493-Walid-BENBIHI.zip  YOLO_small.ckpt  config.py   filters.py                          hourglass_tiny.py     opencv_ffmpeg_64.dll  predictClass.py  v0.1
README.md                                  __pycache__      datagen.py  hg_refined_200.data-00000-of-00001  inference.py          p4frames.mat          timer.py         yolo_net.py
[jalal@goku hourglasstensorlfow]$ /scratch/sjn/anaconda/bin/python  predictClass.py 
/scratch/sjn/anaconda/lib/python3.6/site-packages/h5py/__init__.py:36: FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
  from ._conv import register_converters as _register_converters
Traceback (most recent call last):
  File "predictClass.py", line 1283, in <module>
    predict = PredictProcessor(params)
  File "predictClass.py", line 63, in __init__
    self.HG = HourglassModel(nFeat= self.params['nfeats'], nStack = self.params['nstacks'], 
KeyError: 'nfeats'
[jalal@goku hourglasstensorlfow]$