Closed yashiro32 closed 6 years ago
Did you first run the setup.sh file?
Yes, I ran the setup.sh to download the trained models and to install the external utilities.
The scripts works on tensorflow-gpu but not on tensorflow-cpu and gotten the error I stated previously.
Ohh well..!! I ran it on tensorflow-gpu. So I guess I didnt get any error. The script works perfectly well but takes more time. I am trying to optimize it.
It's on tensorflow-cpu that is getting the error. I ran it on cpu because the scripts need 4.9 GB of memory to run and my the memory on my gpu is insufficient.
I've uploaded a faster version that should also require less memory in the GPU. If that it's still not enough, you can try to reduce the batch size in the file config.py which would reduce further the space needed to allocate the graph.
p.s. The network is not very fast if the code is run on a single image. That it's tf problem that has to initialize some stuff the first time you call the pose_estimator.estimate(image)
method. If this is executed on other images then it runs at 0.34 secs per frame on a Titan X.
setup.sh can be run from git bash on windows.
or
Download the *.tar.tz files manually by pasting the URLs from the setup.sh files and excecute each instruction line by line in git bash.
This is how I overcame the same issue
I got this error: "Segmentation fault (core dumped)" when running your scripts on the cpu.