Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Mediapipe pose use instead of openpose #4

Closed
1kaiser opened this issue May 17, 2021 · 10 comments
Closed

Mediapipe pose use instead of openpose #4

1kaiser opened this issue May 17, 2021 · 10 comments

Comments

@1kaiser
Copy link

1kaiser commented May 17, 2021

Media pipe pose use instead of open pose because it throws building error

https://colab.research.google.com/drive/1uCuA6We9T5r0WljspEHWPHXCT_2bMKUy

https://google.github.io/mediapipe/solutions/holistic

@sergeyprokudin
Copy link
Owner

Hi,

I'll add a workaround for this in the next few days. Until that, you can use the following solution and copy the necessary OpenPose models' files from the Kaggle repository:

@1kaiser 1kaiser closed this as completed May 17, 2021
@1kaiser
Copy link
Author

1kaiser commented Jun 5, 2021

@sergeyprokudin can you please make a video running the Google colaboratory versions and post it on you tube. So that the process where we get struck.. we can replicate your steps for overcoming the problems. Screen capture video.

@1kaiser 1kaiser reopened this Jun 5, 2021
@sergeyprokudin
Copy link
Owner

Hi @1kaiser! I might add it later, but for now, you can simply tell me at which cell you start experiencing problems. I just checked the latest Colab demo and it runs OK.

@1kaiser
Copy link
Author

1kaiser commented Jun 6, 2021

Okay, i will try.

@1kaiser 1kaiser closed this as completed Jun 6, 2021
@1kaiser
Copy link
Author

1kaiser commented Jun 6, 2021

error from last cell

aN loss value, stopping! Camera initialization done after 1.2331 Processing: /content/data/images/unscreen-144.png Found Trained Model: /content/vposer/vposer_v1_0/snapshots/TR00_E096.pt NaN loss value, stopping! Camera initialization done after 1.2303 Processing the data took: 00 hours, 03 minutes, 03 seconds

_ after running this ::_

`# @title Make train-test-validation splits, copy data to final folders

import shutil

train_ratio = 0.9 #@param

final_zip_path = '/content/gdrive/MyDrive/PRINT/G/kaiser/images/smplpix_data_test.zip' # @param {type:"string"}

target_images_path = '/content/data/smplifyx_results/input_images'
smplifyx_renders = '/content/data/smplifyx_results/rendered_smplifyx_meshes'

smplpix_data_path = '/content/smplpix_data'

train_input_dir = os.path.join(smplpix_data_path, 'train', 'input')
train_output_dir = os.path.join(smplpix_data_path, 'train', 'output')
val_input_dir = os.path.join(smplpix_data_path, 'validation', 'input')
val_output_dir = os.path.join(smplpix_data_path, 'validation', 'output')
test_input_dir = os.path.join(smplpix_data_path, 'test', 'input')
test_output_dir = os.path.join(smplpix_data_path, 'test', 'output')

!mkdir -p $train_input_dir
!mkdir -p $train_output_dir
!mkdir -p $val_input_dir
!mkdir -p $val_output_dir
!mkdir -p $test_input_dir
!mkdir -p $test_output_dir

img_names = sorted(os.listdir(target_images_path))
n_images = len(img_names)
n_train_images = int(n_images * train_ratio)
n_val_images = int(n_images * (1-train_ratio) / 2)
train_images = img_names[0:n_train_images]
val_images = img_names[n_train_images:n_train_images+n_val_images]
test_images = img_names[n_train_images:]

for img in train_images:
shutil.copy(os.path.join(smplifyx_renders, img), train_input_dir)
shutil.copy(os.path.join(target_images_path, img), train_output_dir)

for img in val_images:
shutil.copy(os.path.join(smplifyx_renders, img), val_input_dir)
shutil.copy(os.path.join(target_images_path, img), val_output_dir)

for img in test_images:
shutil.copy(os.path.join(smplifyx_renders, img), test_input_dir)
shutil.copy(os.path.join(target_images_path, img), test_output_dir)

%cd /content
!zip -r $final_zip_path smplpix_data/`

error

FileNotFoundError                         Traceback (most recent call last)
<ipython-input-24-397f5e2204f0> in <module>()
     27 get_ipython().system('mkdir -p $test_output_dir')
     28 






---> 29 img_names = sorted(os.listdir(target_images_path))
     30 n_images = len(img_names)
     31 n_train_images = int(n_images * train_ratio)

FileNotFoundError: [Errno 2] No such file or directory: '/content/data/smplifyx_results/input_images'

@1kaiser 1kaiser reopened this Jun 6, 2021
@sergeyprokudin
Copy link
Owner

error from last cell

aN loss value, stopping! Camera initialization done after 1.2331 Processing: /content/data/images/unscreen-144.png Found Trained Model: /content/vposer/vposer_v1_0/snapshots/TR00_E096.pt NaN loss value, stopping! Camera initialization done after 1.2303 Processing the data took: 00 hours, 03 minutes, 03 seconds

_ after running this ::_

`# @title Make train-test-validation splits, copy data to final folders

import shutil

train_ratio = 0.9 #@param

final_zip_path = '/content/gdrive/MyDrive/PRINT/G/kaiser/images/smplpix_data_test.zip' # @param {type:"string"}

target_images_path = '/content/data/smplifyx_results/input_images'
smplifyx_renders = '/content/data/smplifyx_results/rendered_smplifyx_meshes'

smplpix_data_path = '/content/smplpix_data'

train_input_dir = os.path.join(smplpix_data_path, 'train', 'input')
train_output_dir = os.path.join(smplpix_data_path, 'train', 'output')
val_input_dir = os.path.join(smplpix_data_path, 'validation', 'input')
val_output_dir = os.path.join(smplpix_data_path, 'validation', 'output')
test_input_dir = os.path.join(smplpix_data_path, 'test', 'input')
test_output_dir = os.path.join(smplpix_data_path, 'test', 'output')

!mkdir -p $train_input_dir
!mkdir -p $train_output_dir
!mkdir -p $val_input_dir
!mkdir -p $val_output_dir
!mkdir -p $test_input_dir
!mkdir -p $test_output_dir

img_names = sorted(os.listdir(target_images_path))
n_images = len(img_names)
n_train_images = int(n_images * train_ratio)
n_val_images = int(n_images * (1-train_ratio) / 2)
train_images = img_names[0:n_train_images]
val_images = img_names[n_train_images:n_train_images+n_val_images]
test_images = img_names[n_train_images:]

for img in train_images:
shutil.copy(os.path.join(smplifyx_renders, img), train_input_dir)
shutil.copy(os.path.join(target_images_path, img), train_output_dir)

for img in val_images:
shutil.copy(os.path.join(smplifyx_renders, img), val_input_dir)
shutil.copy(os.path.join(target_images_path, img), val_output_dir)

for img in test_images:
shutil.copy(os.path.join(smplifyx_renders, img), test_input_dir)
shutil.copy(os.path.join(target_images_path, img), test_output_dir)

%cd /content
!zip -r $final_zip_path smplpix_data/`

error

FileNotFoundError                         Traceback (most recent call last)
<ipython-input-24-397f5e2204f0> in <module>()
     27 get_ipython().system('mkdir -p $test_output_dir')
     28 






---> 29 img_names = sorted(os.listdir(target_images_path))
     30 n_images = len(img_names)
     31 n_train_images = int(n_images * train_ratio)

FileNotFoundError: [Errno 2] No such file or directory: '/content/data/smplifyx_results/input_images'

this looks like a SMPLify-X error. Are you experiencing it when running the code on demo data or your own?

@1kaiser
Copy link
Author

1kaiser commented Jun 6, 2021

Own data

@sergeyprokudin
Copy link
Owner

I would recommend then creating an issue at the SMPLify-X repository. It is highly likely that the optimisation is failing on your input frames for some reason (e.g. bad OpenPose detections).

@1kaiser
Copy link
Author

1kaiser commented Jun 6, 2021

Ok i will try with high quality data, thanks 🙏🏻

@ss8319
Copy link

ss8319 commented Nov 28, 2022

Hi,

I'll add a workaround for this in the next few days. Until that, you can use the following solution and copy the necessary OpenPose models' files from the Kaggle repository:

Hi @sergeyprokudin @1kaiser

Did you manage to get MediaPipe connected? Mind sharing your Colab notebook?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants