search

vchoutas / smplify-x

Expressive Body Capture: 3D Hands, Face, and Body from a Single Image
https://smpl-x.is.tue.mpg.de/
Other
1.77k stars 340 forks source link

CUDA 10.0 #12

Closed Piero-M closed 4 years ago

Piero-M commented 5 years ago

Sorry, but the program does not work without CUDA? I would like to try it on Mac with CPU

vchoutas commented 5 years ago

@PieroMondora you can set use_cuda=False and interpenetration=False.

Piero-M commented 5 years ago

@vchoutas I did, but he's giving me this problem:

AttributeError: module 'networkx' has no attribute 'DiGraph'

vchoutas commented 5 years ago

@PieroMondora Can you please share the full error message?

Piero-M commented 5 years ago

@vchoutas

this is the command: python smplifyx/main.py --config cfg_files/fit_smplx.yaml --data_folder data --output_folder output --visualize="True" --model_folder models --vposer_ckpt models/vposer_v1_0

this is the error:

Processing: data/images/COCO_val2014_000000000459.jpg
Found Trained Model: models/vposer_v1_0/snapshots/TR00_E096.pt
searching for scad in: /Users/piero/anaconda3/envs/Fiveer_02_06_19/bin:/Users/piero/anaconda3/condabin:/Users/pieromondora/anaconda3/bin:/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin:/Applications/VMware Fusion.app/Contents/Public:/Applications/OpenSCAD.app/Contents/MacOS
Traceback (most recent call last):
  File "smplifyx/main.py", line 272, in <module>
    main(**args)
  File "smplifyx/main.py", line 262, in main
    **args)
  File "/Users/piero/Desktop/Fiverr_02_06_19/smplify-x/smplifyx/fit_single_frame.py", line 306, in fit_single_frame
    batch_size=batch_size, visualize=visualize, **kwargs) as monitor:
  File "/Users/piero/Desktop/Fiverr_02_06_19/smplify-x/smplifyx/fitting.py", line 133, in __enter__
    self.mv = MeshViewer(body_color=self.body_color)
  File "/Users/piero/Desktop/Fiverr_02_06_19/smplify-x/smplifyx/mesh_viewer.py", line 36, in __init__
    import trimesh
  File "/Users/piero/anaconda3/envs/Fiveer_02_06_19/lib/python3.6/site-packages/trimesh/__init__.py", line 15, in <module>
    from .base import Trimesh
  File "/Users/piero/anaconda3/envs/Fiveer_02_06_19/lib/python3.6/site-packages/trimesh/base.py", line 46, in <module>
    from .scene import Scene
  File "/Users/piero/anaconda3/envs/Fiveer_02_06_19/lib/python3.6/site-packages/trimesh/scene/__init__.py", line 3, in <module>
    from .scene import Scene, split_scene
  File "/Users/piero/anaconda3/envs/Fiveer_02_06_19/lib/python3.6/site-packages/trimesh/scene/scene.py", line 19, in <module>
    from .transforms import TransformForest
  File "/Users/piero/anaconda3/envs/Fiveer_02_06_19/lib/python3.6/site-packages/trimesh/scene/transforms.py", line 371, in <module>
    class EnforcedForest(nx.DiGraph):
AttributeError: module 'networkx' has no attribute 'DiGraph'
vchoutas commented 5 years ago

@PieroMondora can you use trimesh by itself? Some script to load a test mesh?

Piero-M commented 5 years ago

@vchoutas I don't know what I'm supposed to do.

vchoutas commented 5 years ago

@PieroMondora could you try re-installing or updating trimesh to see if this fixes the issue?