|
|
|
@ -466,6 +466,7 @@ |
|
|
|
|
" #gitclone(\"https://github.com/facebookresearch/SLIP.git\")\n", |
|
|
|
|
" gitclone(\"https://github.com/crowsonkb/guided-diffusion\")\n", |
|
|
|
|
" gitclone(\"https://github.com/assafshocher/ResizeRight.git\")\n", |
|
|
|
|
" gitclone(\"https://github.com/MSFTserver/pytorch3d-lite.git\")\n", |
|
|
|
|
" pipie(\"./CLIP\")\n", |
|
|
|
|
" pipie(\"./guided-diffusion\")\n", |
|
|
|
|
" multipip_res = subprocess.run(['pip', 'install', 'lpips', 'datetime', 'timm', 'ftfy'], stdout=subprocess.PIPE).stdout.decode('utf-8')\n", |
|
|
|
@ -491,19 +492,8 @@ |
|
|
|
|
"import sys\n", |
|
|
|
|
"import torch\n", |
|
|
|
|
"\n", |
|
|
|
|
"#Install pytorch3d\n", |
|
|
|
|
"if is_colab:\n", |
|
|
|
|
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n", |
|
|
|
|
" version_str=\"\".join([\n", |
|
|
|
|
" f\"py3{sys.version_info.minor}_cu\",\n", |
|
|
|
|
" torch.version.cuda.replace(\".\",\"\"),\n", |
|
|
|
|
" f\"_pyt{pyt_version_str}\"\n", |
|
|
|
|
" ])\n", |
|
|
|
|
" multipip_res = subprocess.run(['pip', 'install', 'fvcore', 'iopath'], stdout=subprocess.PIPE).stdout.decode('utf-8')\n", |
|
|
|
|
" print(multipip_res)\n", |
|
|
|
|
" subprocess.run(['pip', 'install', '--no-index', '--no-cache-dir', 'pytorch3d', '-f', f'https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html'], stdout=subprocess.PIPE).stdout.decode('utf-8')\n", |
|
|
|
|
"\n", |
|
|
|
|
"# sys.path.append('./SLIP')\n", |
|
|
|
|
"sys.path.append('./pytorch3d-lite')\n", |
|
|
|
|
"sys.path.append('./ResizeRight')\n", |
|
|
|
|
"sys.path.append('./MiDaS')\n", |
|
|
|
|
"from dataclasses import dataclass\n", |
|
|
|
@ -725,7 +715,7 @@ |
|
|
|
|
"\n", |
|
|
|
|
"# https://gist.github.com/adefossez/0646dbe9ed4005480a2407c62aac8869\n", |
|
|
|
|
"\n", |
|
|
|
|
"import pytorch3d.transforms as p3dT\n", |
|
|
|
|
"import py3d_tools as p3dT\n", |
|
|
|
|
"import disco_xform_utils as dxf\n", |
|
|
|
|
"\n", |
|
|
|
|
"def interp(t):\n", |
|
|
|
|