the problem when I use hps_type pixie about icon

[YuliangXiu]

Please use PyYAML>=6.0

[TxT1212]

hello, I met the same error, and I have pyyaml 6.0

$ pip install pyyaml
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Requirement already satisfied: pyyaml in /home/txt/anaconda3/envs/icon/lib/python3.8/site-packages (6.0)

any suggestions?

[YuliangXiu]

Sorry, try to use PyYAML==5.1.1

[TxT1212]

Thanks, there are still some conflicts

UnsatisfiableError: The following specifications were found
to be incompatible with the existing python installation in your environment:
Specifications:
pyyaml==5.1.1 -> python[version='>=2.7,<2.8.0a0|>=3.6,<3.7.0a0|>=3.7,<3.8.0a0']
Your python: python=3.8

installion.md requires Python = 3.8.
I'm trying to solve it.

[pinga999]

Sorry, try to use PyYAML==5.1.1

Traceback (most recent call last):
  File "infer.py", line 96, in <module>
    dataset = TestDataset(dataset_param, device)
  File "/workspace/fashion-ICON/apps/../lib/dataset/TestDataset.py", line 105, in __init__
    self.hps = PIXIE(config = pixie_cfg, device=self.device)
  File "/workspace/fashion-ICON/apps/../lib/pixielib/pixie.py", line 49, in __init__
    self._create_model()
  File "/workspace/fashion-ICON/apps/../lib/pixielib/pixie.py", line 115, in _create_model
    self.smplx = SMPLX(self.cfg.model).to(self.device)
  File "/workspace/fashion-ICON/apps/../lib/pixielib/models/SMPLX.py", line 156, in __init__
    self.extra_joint_selector = JointsFromVerticesSelector(
  File "/workspace/fashion-ICON/apps/../lib/pixielib/models/lbs.py", line 399, in __init__
    data = yaml.load(f)
TypeError: load() missing 1 required positional argument: 'Loader'

by original repo's issue (#30)

refer to upper comments, this error might be resolved by install PyYAML==5.1.1 but
It makes error again

Traceback (most recent call last):
  File "infer.py", line 102, in <module>
    for data in pbar:
  File "/opt/conda/envs/icon/lib/python3.8/site-packages/tqdm/std.py", line 1180, in __iter__
    for obj in iterable:
  File "/workspace/fashion-ICON/apps/../lib/dataset/TestDataset.py", line 191, in __getitem__
    preds_dict = self.hps.forward(img_hps.to(self.device))
  File "/workspace/fashion-ICON/apps/../lib/pixielib/pixie.py", line 56, in forward
    param_dict = self.encode({'body': {'image': data}}, threthold=True, keep_local=True, copy_and_paste=False)
  File "/opt/conda/envs/icon/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context
    return func(*args, **kwargs)
  File "/workspace/fashion-ICON/apps/../lib/pixielib/pixie.py", line 259, in encode
    cropped_image, cropped_joints_dict = self.part_from_body(image_hd, part_name, points_dict)
  File "/workspace/fashion-ICON/apps/../lib/pixielib/pixie.py", line 166, in part_from_body
    cropped_image, tform = self.Cropper[cropper_key].crop(
  File "/workspace/fashion-ICON/apps/../lib/pixielib/utils/tensor_cropper.py", line 98, in crop
    cropped_image, tform = crop_tensor(image, center, bbox_size, self.crop_size)
  File "/workspace/fashion-ICON/apps/../lib/pixielib/utils/tensor_cropper.py", line 78, in crop_tensor
    cropped_image = warp_affine(
TypeError: warp_affine() got an unexpected keyword argument 'flags'

Are there any solution to run pixie module?