Merge pull request #3 from gopinath1509/main

Update dependencies and add missing files

environment.yml

@@ -12,13 +12,16 @@ dependencies:
  - plyfile
  - pre-commit
  - pyg::pytorch-scatter
- - pytorch::pytorch
- - pytorch::pytorch-cuda=11.7
+ - pytorch::pytorch=2.*=*cuda*
+ - pytorch::pytorch-cuda=11.8
  - scipy
  - scikit-learn
  - tqdm
  - trimesh
+ - wandb
+ - loguru
  - pip:
+ - yacs
  - smplx
  - webdataset
  - -e external/vgtk/vgtk

external/vgtk/vgtk/data/anchors/kpsphere24.ply

@@ -0,0 +1,35 @@
+ply
+format ascii 1.0
+element vertex 24
+property float x
+property float y
+property float z
+property uchar red
+property uchar green
+property uchar blue
+end_header
+0.0000 0.0000 0.0000 255 255 255 
+0.6299 -0.2149 -0.7423 255 255 255 
+-0.3022 0.7119 -0.6461 255 255 255 
+-0.2966 -0.9471 -0.1752 255 255 255 
+0.2247 0.2260 0.9405 255 255 255 
+-0.9917 0.0428 0.1434 255 255 255 
+0.5258 0.7599 0.4020 255 255 255 
+-0.2035 -0.0206 -0.9718 255 255 255 
+0.3777 0.4709 -0.8007 255 255 255 
+0.0201 -0.8949 0.4631 255 255 255 
+0.1504 0.9692 -0.1638 255 255 255 
+-0.6206 0.2209 0.7483 255 255 255 
+-0.7326 -0.4753 -0.4757 255 255 255 
+-0.1841 0.7913 0.5883 255 255 255 
+-0.7168 -0.5765 0.3743 255 255 255 
+0.5800 -0.4954 0.6381 255 255 255 
+0.0465 -0.6561 -0.7571 255 255 255 
+0.8635 0.1413 0.4999 255 255 255 
+0.5038 -0.8428 -0.1487 255 255 255 
+-0.6553 0.7446 0.0504 255 255 255 
+0.9561 -0.2905 -0.0871 255 255 255 
+-0.8158 0.2253 -0.5471 255 255 255 
+-0.2005 -0.3558 0.9158 255 255 255 
+0.8413 0.4655 -0.2484 255 255 255 
+

external/vgtk/vgtk/data/anchors/kpsphere30.ply

@@ -0,0 +1,41 @@
+ply
+format ascii 1.0
+element vertex 30
+property float x
+property float y
+property float z
+property uchar red
+property uchar green
+property uchar blue
+end_header
+0.0000 0.0000 0.0000 255 255 255 
+0.3652 -0.3791 -0.8628 255 255 255 
+-0.4803 0.6644 0.5948 255 255 255 
+0.3332 -0.4217 0.8705 255 255 255 
+-0.6308 0.8077 -0.0669 255 255 255 
+0.1518 -0.9318 0.3939 255 255 255 
+-1.0532 0.2475 0.1430 255 255 255 
+0.3315 -0.9158 -0.2675 255 255 255 
+1.0801 0.0438 0.1552 255 255 255 
+-0.8004 0.3207 -0.6279 255 255 255 
+0.7509 -0.5757 0.3404 255 255 255 
+0.1998 0.3141 -0.9507 255 255 255 
+0.7376 0.7613 0.2401 255 255 255 
+-0.5152 -0.0709 -0.0869 255 255 255 
+-0.3344 -0.0976 -0.9577 255 255 255 
+0.4523 0.8460 -0.4607 255 255 255 
+0.0082 1.0084 0.0926 255 255 255 
+0.8411 -0.4832 -0.3561 255 255 255 
+-0.6912 0.0280 0.7590 255 255 255 
+0.4739 0.2228 -0.0761 255 255 255 
+-0.2315 0.7470 -0.6350 255 255 255 
+-0.1966 -0.7407 -0.6733 255 255 255 
+-0.8860 -0.5346 0.2965 255 255 255 
+-0.3921 -0.9375 -0.0178 255 255 255 
+0.6689 0.1576 0.7531 255 255 255 
+0.8571 0.2060 -0.6344 255 255 255 
+-0.3328 -0.5827 0.7463 255 255 255 
+-0.0618 0.1242 1.0014 255 255 255 
+-0.8578 -0.4656 -0.4937 255 255 255 
+0.1931 0.6978 0.7148 255 255 255 
+

external/vgtk/vgtk/data/anchors/kpsphere66.ply

@@ -0,0 +1,77 @@
+ply
+format ascii 1.0
+element vertex 66
+property float x
+property float y
+property float z
+property uchar red
+property uchar green
+property uchar blue
+end_header
+0.0000 0.0000 0.0000 255 255 255 
+0.0000 0.0000 0.5000 255 255 255 
+0.4330 0.0000 0.2500 255 255 255 
+0.4330 0.0000 -0.2500 255 255 255 
+-0.0000 -0.0000 -0.5000 255 255 255 
+-0.0000 0.0000 0.5000 255 255 255 
+-0.0000 0.4330 0.2500 255 255 255 
+-0.0000 0.4330 -0.2500 255 255 255 
+0.0000 -0.0000 -0.5000 255 255 255 
+-0.0000 -0.0000 0.5000 255 255 255 
+-0.4330 -0.0000 0.2500 255 255 255 
+-0.4330 -0.0000 -0.2500 255 255 255 
+0.0000 0.0000 -0.5000 255 255 255 
+0.0000 -0.0000 0.5000 255 255 255 
+0.0000 -0.4330 0.2500 255 255 255 
+0.0000 -0.4330 -0.2500 255 255 255 
+-0.0000 0.0000 -0.5000 255 255 255 
+0.0000 0.0000 1.0000 255 255 255 
+0.5000 0.0000 0.8660 255 255 255 
+0.8660 0.0000 0.5000 255 255 255 
+1.0000 0.0000 -0.0000 255 255 255 
+0.8660 0.0000 -0.5000 255 255 255 
+0.5000 0.0000 -0.8660 255 255 255 
+-0.0000 -0.0000 -1.0000 255 255 255 
+0.0000 0.0000 1.0000 255 255 255 
+0.3117 0.3909 0.8660 255 255 255 
+0.5400 0.6771 0.5000 255 255 255 
+0.6235 0.7818 -0.0000 255 255 255 
+0.5400 0.6771 -0.5000 255 255 255 
+0.3117 0.3909 -0.8660 255 255 255 
+-0.0000 -0.0000 -1.0000 255 255 255 
+-0.0000 0.0000 1.0000 255 255 255 
+-0.1113 0.4875 0.8660 255 255 255 
+-0.1927 0.8443 0.5000 255 255 255 
+-0.2225 0.9749 -0.0000 255 255 255 
+-0.1927 0.8443 -0.5000 255 255 255 
+-0.1113 0.4875 -0.8660 255 255 255 
+0.0000 -0.0000 -1.0000 255 255 255 
+-0.0000 0.0000 1.0000 255 255 255 
+-0.4505 0.2169 0.8660 255 255 255 
+-0.7803 0.3758 0.5000 255 255 255 
+-0.9010 0.4339 -0.0000 255 255 255 
+-0.7803 0.3758 -0.5000 255 255 255 
+-0.4505 0.2169 -0.8660 255 255 255 
+0.0000 -0.0000 -1.0000 255 255 255 
+-0.0000 -0.0000 1.0000 255 255 255 
+-0.4505 -0.2169 0.8660 255 255 255 
+-0.7803 -0.3758 0.5000 255 255 255 
+-0.9010 -0.4339 -0.0000 255 255 255 
+-0.7803 -0.3758 -0.5000 255 255 255 
+-0.4505 -0.2169 -0.8660 255 255 255 
+0.0000 0.0000 -1.0000 255 255 255 
+-0.0000 -0.0000 1.0000 255 255 255 
+-0.1113 -0.4875 0.8660 255 255 255 
+-0.1927 -0.8443 0.5000 255 255 255 
+-0.2225 -0.9749 -0.0000 255 255 255 
+-0.1927 -0.8443 -0.5000 255 255 255 
+-0.1113 -0.4875 -0.8660 255 255 255 
+0.0000 0.0000 -1.0000 255 255 255 
+0.0000 -0.0000 1.0000 255 255 255 
+0.3117 -0.3909 0.8660 255 255 255 
+0.5400 -0.6771 0.5000 255 255 255 
+0.6235 -0.7818 -0.0000 255 255 255 
+0.5400 -0.6771 -0.5000 255 255 255 
+0.3117 -0.3909 -0.8660 255 255 255 
+-0.0000 0.0000 -1.0000 255 255 255 
+

external/vgtk/vgtk/data/anchors/sphere162.ply

@@ -0,0 +1,173 @@
+ply
+format ascii 1.0
+element vertex 162
+property float x
+property float y
+property float z
+property uchar red
+property uchar green
+property uchar blue
+end_header
+0.0000 0.0000 -1.0000 255 255 255 
+0.0000 0.0000 1.0000 255 255 255 
+0.0000 -1.0000 0.0000 255 255 255 
+-0.0000 1.0000 0.0000 255 255 255 
+-1.0000 -0.0000 0.0000 255 255 255 
+1.0000 0.0000 -0.0000 255 255 255 
+0.4360 0.8642 0.2511 255 255 255 
+0.0776 0.2389 0.9679 255 255 255 
+0.0776 -0.2389 0.9679 255 255 255 
+0.2764 -0.8507 0.4472 255 255 255 
+0.2764 0.8507 0.4472 255 255 255 
+-0.7236 -0.5257 0.4472 255 255 255 
+-0.7236 0.5257 0.4472 255 255 255 
+0.2629 0.8090 -0.5257 255 255 255 
+0.2629 -0.8090 -0.5257 255 255 255 
+-0.4472 -0.0000 0.8944 255 255 255 
+-0.4472 -0.5257 -0.7236 255 255 255 
+-0.4472 -0.8507 0.2764 255 255 255 
+-0.4472 0.8507 0.2764 255 255 255 
+-0.4472 0.5257 -0.7236 255 255 255 
+0.9566 0.1476 0.2511 255 255 255 
+0.9566 -0.1476 0.2511 255 255 255 
+0.1552 0.9554 0.2511 255 255 255 
+0.1552 -0.9554 0.2511 255 255 255 
+0.8127 -0.2952 -0.5023 255 255 255 
+0.8127 0.2952 -0.5023 255 255 255 
+0.2328 -0.7166 0.6575 255 255 255 
+0.2328 0.7166 0.6575 255 255 255 
+-0.1625 -0.5000 -0.8507 255 255 255 
+-0.1625 0.5000 -0.8507 255 255 255 
+0.6708 -0.1625 -0.7236 255 255 255 
+0.6708 0.6882 0.2764 255 255 255 
+0.6708 0.1625 -0.7236 255 255 255 
+0.1382 -0.9511 -0.2764 255 255 255 
+0.1382 -0.4253 -0.8944 255 255 255 
+0.1382 0.4253 -0.8944 255 255 255 
+0.1382 0.9511 -0.2764 255 255 255 
+-0.8618 -0.4253 -0.2764 255 255 255 
+-0.8618 0.4253 -0.2764 255 255 255 
+-0.3090 -0.9511 0.0000 255 255 255 
+-0.3090 0.9511 0.0000 255 255 255 
+0.8090 -0.5878 0.0000 255 255 255 
+0.8090 0.5878 0.0000 255 255 255 
+0.3618 0.2629 0.8944 255 255 255 
+0.3618 -0.5878 -0.7236 255 255 255 
+0.3618 0.5878 -0.7236 255 255 255 
+0.3618 -0.2629 0.8944 255 255 255 
+-0.6382 0.2629 -0.7236 255 255 255 
+-0.6382 -0.2629 -0.7236 255 255 255 
+-0.6872 0.6817 0.2511 255 255 255 
+-0.6872 -0.6817 0.2511 255 255 255 
+-0.0296 0.8642 -0.5023 255 255 255 
+-0.0296 -0.8642 -0.5023 255 255 255 
+-0.2511 -0.0000 -0.9679 255 255 255 
+-0.6095 0.4429 0.6575 255 255 255 
+-0.6095 -0.4429 0.6575 255 255 255 
+-0.8311 0.2389 -0.5023 255 255 255 
+-0.8311 -0.2389 -0.5023 255 255 255 
+-0.5319 0.6817 0.5023 255 255 255 
+-0.5319 -0.6817 0.5023 255 255 255 
+-0.7534 -0.0000 -0.6575 255 255 255 
+-0.5257 -0.0000 -0.8507 255 255 255 
+0.4840 -0.7166 0.5023 255 255 255 
+0.4840 0.7166 0.5023 255 255 255 
+0.2032 0.1476 -0.9679 255 255 255 
+0.2032 -0.1476 -0.9679 255 255 255 
+-0.9511 0.3090 0.0000 255 255 255 
+-0.2629 -0.8090 0.5257 255 255 255 
+0.8607 0.4429 -0.2511 255 255 255 
+0.8607 -0.4429 -0.2511 255 255 255 
+-0.6882 -0.5000 -0.5257 255 255 255 
+-0.6882 0.5000 -0.5257 255 255 255 
+0.4360 -0.8642 0.2511 255 255 255 
+-0.4253 -0.3090 0.8507 255 255 255 
+-0.4253 0.3090 0.8507 255 255 255 
+-0.8507 -0.0000 0.5257 255 255 255 
+-0.8944 -0.0000 -0.4472 255 255 255 
+0.6708 -0.6882 0.2764 255 255 255 
+0.9472 -0.1625 -0.2764 255 255 255 
+0.9472 0.1625 -0.2764 255 255 255 
+-0.0528 -0.6882 0.7236 255 255 255 
+-0.0528 0.6882 0.7236 255 255 255 
+-0.5878 0.8090 0.0000 255 255 255 
+-0.5878 -0.8090 0.0000 255 255 255 
+0.5878 0.8090 0.0000 255 255 255 
+0.5878 -0.8090 0.0000 255 255 255 
+0.0528 -0.6882 -0.7236 255 255 255 
+0.0528 0.6882 -0.7236 255 255 255 
+-0.9472 -0.1625 0.2764 255 255 255 
+-0.9472 0.1625 0.2764 255 255 255 
+-0.6708 -0.6882 -0.2764 255 255 255 
+0.8944 0.0000 0.4472 255 255 255 
+0.8507 -0.0000 -0.5257 255 255 255 
+0.4253 -0.3090 -0.8507 255 255 255 
+0.4253 0.3090 -0.8507 255 255 255 
+0.6882 -0.5000 0.5257 255 255 255 
+0.6882 0.5000 0.5257 255 255 255 
+-0.8607 0.4429 0.2511 255 255 255 
+-0.8607 -0.4429 0.2511 255 255 255 
+0.9511 -0.3090 0.0000 255 255 255 
+0.9511 0.3090 0.0000 255 255 255 
+-0.2032 0.1476 0.9679 255 255 255 
+-0.2032 -0.1476 0.9679 255 255 255 
+-0.4840 -0.7166 -0.5023 255 255 255 
+-0.4840 0.7166 -0.5023 255 255 255 
+0.5257 0.0000 0.8507 255 255 255 
+0.7534 0.0000 0.6575 255 255 255 
+0.5319 0.6817 -0.5023 255 255 255 
+0.5319 -0.6817 -0.5023 255 255 255 
+0.8311 0.2389 0.5023 255 255 255 
+0.8311 -0.2389 0.5023 255 255 255 
+0.6095 0.4429 -0.6575 255 255 255 
+0.6095 -0.4429 -0.6575 255 255 255 
+0.2511 0.0000 0.9679 255 255 255 
+0.0296 0.8642 0.5023 255 255 255 
+0.0296 -0.8642 0.5023 255 255 255 
+0.6872 0.6817 -0.2511 255 255 255 
+0.6872 -0.6817 -0.2511 255 255 255 
+0.6382 0.2629 0.7236 255 255 255 
+0.6382 -0.2629 0.7236 255 255 255 
+-0.3618 0.2629 -0.8944 255 255 255 
+-0.3618 -0.5878 0.7236 255 255 255 
+-0.3618 0.5878 0.7236 255 255 255 
+-0.3618 -0.2629 -0.8944 255 255 255 
+-0.8090 -0.5878 -0.0000 255 255 255 
+-0.8090 0.5878 -0.0000 255 255 255 
+-0.9511 -0.3090 0.0000 255 255 255 
+0.3090 -0.9511 -0.0000 255 255 255 
+0.3090 0.9511 -0.0000 255 255 255 
+0.8618 -0.4253 0.2764 255 255 255 
+0.8618 0.4253 0.2764 255 255 255 
+-0.1382 -0.9511 0.2764 255 255 255 
+-0.1382 -0.4253 0.8944 255 255 255 
+-0.1382 0.4253 0.8944 255 255 255 
+-0.1382 0.9511 0.2764 255 255 255 
+-0.6708 -0.1625 0.7236 255 255 255 
+-0.6708 0.6882 -0.2764 255 255 255 
+-0.6708 0.1625 0.7236 255 255 255 
+0.1625 -0.5000 0.8507 255 255 255 
+0.1625 0.5000 0.8507 255 255 255 
+-0.2328 -0.7166 -0.6575 255 255 255 
+-0.2328 0.7166 -0.6575 255 255 255 
+-0.8127 -0.2952 0.5023 255 255 255 
+-0.8127 0.2952 0.5023 255 255 255 
+-0.1552 0.9554 -0.2511 255 255 255 
+-0.1552 -0.9554 -0.2511 255 255 255 
+-0.9566 0.1476 -0.2511 255 255 255 
+-0.9566 -0.1476 -0.2511 255 255 255 
+0.4472 -0.0000 -0.8944 255 255 255 
+0.4472 -0.5257 0.7236 255 255 255 
+0.4472 -0.8507 -0.2764 255 255 255 
+0.4472 0.8507 -0.2764 255 255 255 
+0.4472 0.5257 0.7236 255 255 255 
+-0.2629 0.8090 0.5257 255 255 255 
+0.7236 -0.5257 -0.4472 255 255 255 
+0.7236 0.5257 -0.4472 255 255 255 
+-0.2764 -0.8507 -0.4472 255 255 255 
+-0.2764 0.8507 -0.4472 255 255 255 
+-0.0776 0.2389 -0.9679 255 255 255 
+-0.0776 -0.2389 -0.9679 255 255 255 
+-0.4360 0.8642 -0.2511 255 255 255 
+-0.4360 -0.8642 -0.2511 255 255 255 
+