import tensorflow as tf
from tf_smpl import SMPL
from tf_smgl import SMGL

smpl = SMPL("<path-to-smpl:pkl>")
smgl = SMGL("<path-to-smgl:pkl>")
# calculate SMPL vertices
v_body, body_dict = smpl(
  shapes=betas,
  poses=poses,
  trans=trans,
  includes=["J_transforms"]
)
# calculate body normals
v_normals = smpl.normals(v_body)
# calculate SMGL vertices
v_garment = smgl(
  shapes=betas, 
  poses=poses, 
  trans=trans, 
  body_dict=body_dict
)
# Calculate neighbours of SMPL with Outfit
v_indices = smpl.neighbours(v_body, v_garment)
# Fix collision errors
v_fixed = smgl.fix_collisions(v_garment, v_body, v_normals, v_indices=v_indices)