shunsukesaito/PIFu

sh ./scripts/test.sh error

RyanG41 opened this issue · 2 comments

Either in Colab or local enviroment the erorr shows up anyway. Please let me know how to fix this. Thanks a lot.

sh ./scripts/test.sh
`+ GPU_ID=0

  • DISPLAY_ID=10
  • NAME=pifu_demo
  • BATCH_SIZE=1
  • MLP_DIM=257 1024 512 256 128 1
  • MLP_DIM_COLOR=513 1024 512 256 128 3
  • VOL_RES=256
  • CHECKPOINTS_NETG_PATH=./checkpoints/net_G
  • CHECKPOINTS_NETC_PATH=./checkpoints/net_C
  • TEST_FOLDER_PATH=./sample_images
  • CUDA_VISIBLE_DEVICES=0 python ./apps/eval.py --name pifu_demo --batch_size 1 --mlp_dim 257 1024 512 256 128 1 --mlp_dim_color 513 1024 512 256 128 3 --num_stack 4 --num_hourglass 2 --resolution 256 --hg_down ave_pool --norm group --norm_color group --test_folder_path ./sample_images --load_netG_checkpoint_path ./checkpoints/net_G --load_netC_checkpoint_path ./checkpoints/net_C
    initialize network with normal
    Using Network: hgpifu
    loading for net C ... ./checkpoints/net_C
    initialize network with normal
    num; 1
    0it [00:00, ?it/s]./sample_images/ryota.png ./sample_images/ryota_mask.png
    error cannot marching cubes
    cannot unpack non-iterable int object
    Can not create marching cubes at this time.

    1it [00:12, 12.10s/it]`

Had the same issue, fixed by downgrading scikit-image to 0.18.3 (python3.9)

change measure.marching_cubes_lewiner to measure.marching_cubes would work for me, you can try this