initial app test

app.py

@@ -9,7 +9,6 @@ import sys
 print(os.path.abspath(__file__))
 os.system('pip install gradio==3.47.1')
 os.system('pip install networkx')
-os.system('sh fetch_data.sh')
 
 import gradio as gr
 
@@ -137,13 +136,13 @@ def create_scene(mesh, img, focal_length=500, camera_center=250, img_res=500):
     IMG.thumbnail((3000, 3000))
     return IMG    
 
-def main(args, img_src, out_dir, mesh_colour=[130, 130, 130, 255], annot_colour=[0, 255, 0, 255]):
+def main(img_src, out_dir, model_path='checkpoint/deco_best.pth', mesh_colour=[130, 130, 130, 255], annot_colour=[0, 255, 0, 255]):
     if os.path.isdir(img_src):
         images = glob.iglob(img_src + '/*', recursive=True)
     else:
         images = [img_src]
 
-    deco_model = initiate_model(args)
+    deco_model = initiate_model(model_path)
     
     smpl_path = os.path.join(constants.SMPL_MODEL_DIR, 'smpl_neutral_tpose.ply')
     

checkpoint/deco_best.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1d1502292154e446457131dc9915bac15bf530b3e973c226f748417be50601bc
+size 1560039945

data/J_regressor_extra.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cc968ea4f9855571e82f90203280836b01f13ee42a8e1b89d8d580b801242a89
+size 496160

data/conversions/smpl_to_smplx.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e8917053cc980c1218453a563378c9e7d277e71c49def3a26ab72afb19c7b994
+size 578366660

data/conversions/smplx_to_smpl.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:685d935cdf65c44534ec3effdb7753a5886bebd7a1c8f295d55e16e4d1be64fb
+size 578019257

data/smpl/smpl_neutral_geodesic_dist.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b564cd34bf227e41ecdb0186ed5355cb9c4713eb69176bfa80a85bc8edd2bd05
+size 379776928

data/smpl/smplpix_vertex_colors.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a02dd70b033c06f62413a03983476a74c03194edcf204ea0bd20ac650885ab28
+size 220608

data/smpl_partSegmentation_mapping.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2f7f8f5e381b7a92f40964315858492fe0d63e4a27789882fada17a144da800a
+size 56220

data/weights/pose_hrnet_w32_256x192.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:19bc083708bb8d873211e50d85d56344c10290c6e8b564c813fdde09645c4c1c
+size 114721619