change logs

PIFu/apps/eval_spaces.py

@@ -38,7 +38,7 @@ class Evaluator:
         ])
         # set cuda
         cuda = torch.device('cuda:%d' % opt.gpu_id) if torch.cuda.is_available() else torch.device('cpu')
-        print("CUDDAAAAA ???",  torch.cuda.get_device_name(0))
+        print("CUDDAAAAA ???",  torch.cuda.get_device_name(0) if torch.cuda.is_available()  else "NO ONLY CPU")
   
         # create net
         netG = HGPIFuNet(opt, projection_mode).to(device=cuda)

app.py

@@ -1,5 +1,9 @@
 import os
-os.system("pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html")
+try: 
+    os.system("pip install --upgrade  torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html")
+except Exception as e:
+    print(e)
+    
 from pydoc import describe
 from huggingface_hub import hf_hub_download
 import gradio as gr
@@ -30,21 +34,20 @@ def process(img_path):
     print("image name", img_name)
     img = Image.open(img_path)
     # remove background
-    print("remove background")
+    print("Removeing background")
     foreground = Image.fromarray(remove_bg.inference(img), 'RGBA')
     foreground.save("./PIFu/inputs/" + img_name + ".png")
-    print("align mask with input training image")
+    print("Aliging mask with input training image")
     subprocess.Popen(["python", "./apps/crop_img.py", "--input_image",
                       f'./inputs/{img_name}.png', "--out_path", "./inputs"], cwd="PIFu").communicate()
 
-    print("generate 3D model")
+    print("Generating 3D model")
     subprocess.Popen("./scripts/test.sh", env={
                      **env,
                      "INPUT_IMAGE_PATH": f'./inputs/{img_name}.png',
                      "VOL_RES": "128"},
                      cwd="PIFu").communicate()
-
-    print("inference")
+    print("DONE 3D model")
     return f'./PIFu/results/spaces_demo/result_{img_name}.glb'
 
 

requirements.txt

@@ -17,4 +17,6 @@ Shapely==1.7.0
 six==1.14.0
 tqdm==4.43.0
 trimesh==3.5.23
-xxhash==1.4.3
+xxhash==1.4.3
+torch==1.11
+torchvision==0.12.0