update

app.py

@@ -196,7 +196,8 @@ t_list = os.listdir("./nuevostrees/" )
 dataset = tDataset(t_list, "./nuevostrees/")
 data_loader = DataLoader(dataset, batch_size = batch_size, shuffle=True, collate_fn=my_collate)
 
-def predict2():
+
+def predict():
     success = False
     while not success:
         G = nx.Graph()
@@ -285,20 +286,10 @@ def predict2():
 
     minimo = min(min(x), min(y), min(z))
     maximo = max(max(x), max(y), max(z))
-    
-    v_matrix = v_matrix[:50]
-    f_matrix = f_matrix[:100]
-
-    x = x.astype(np.float32)
-    y = y.astype(np.float32)
-    z = z.astype(np.float32)
-    i = f_matrix[:, 0].astype(np.int32)
-    j = f_matrix[:, 1].astype(np.int32)
-    k = f_matrix[:, 2].astype(np.int32)
-
+  
     fig = go.Figure(go.Mesh3d(
             x=x, y=y, z=z, 
-            i=i, j=j, k=k, 
+            i=f_matrix[:,0], j=f_matrix[:,1], k=f_matrix[:,2], 
             color='red'))
     
     fig.update_layout(
@@ -319,36 +310,8 @@ def predict2():
 
     
     #o3d.io.write_triangle_mesh("generadas/"+filename.split("_")[0]+".obj", mesh_o3d)
-    print("Done!")
     return fig
 
-def predict():
-    # Mini esfera
-    phi = np.linspace(0, np.pi, 20)
-    theta = np.linspace(0, 2 * np.pi, 40)
-    phi, theta = np.meshgrid(phi, theta)
-    r = 1
-
-    x = r * np.sin(phi) * np.cos(theta)
-    y = r * np.sin(phi) * np.sin(theta)
-    z = r * np.cos(phi)
-
-    x = x.flatten().astype(np.float32)
-    y = y.flatten().astype(np.float32)
-    z = z.flatten().astype(np.float32)
-
-    # Caras dummy
-    i = np.arange(len(x) - 2).astype(np.int32)
-    j = i + 1
-    k = i + 2
-
-    fig = go.Figure(go.Mesh3d(
-        x=x, y=y, z=z,
-        i=i, j=j, k=k,
-        color='red'
-    ))
-
-    return fig
 
 
 a = [1.,1.,1.]
@@ -365,4 +328,5 @@ Grassdecoder.load_state_dict(checkpoint['decoder_state_dict'])
 gr.Interface( predict, 
     inputs=None,
     outputs=gr.Plot(),
-).launch()
+).launch() #gr.Plot()
+

requirements.txt

@@ -5,6 +5,6 @@ open3d
 matplotlib
 numpy
 scipy
-plotly
+plotly==6.0.1
 pymeshlab
-vedo
+gradio==5.24.0

sinsub.obj

@@ -2,74 +2,74 @@
 # object name: sinsub
 # number of vertices: 68
 # number of triangles: 126
-v -0.013401 0.395188 0.516043
-v -0.0196831 0.417634 0.492699
-v 0.0124926 0.419258 0.485602
-v 0.0187747 0.396812 0.508947
-v 0.141308 0.251278 0.619399
-v 0.178805 0.264909 0.629869
-v 0.195202 0.244072 0.598272
-v 0.157705 0.230441 0.587802
-v 0.0493672 0.212719 0.688877
-v 0.0901493 0.213399 0.715386
-v 0.108316 0.177257 0.688366
-v 0.067534 0.176577 0.661857
-v 0.0108628 0.157246 0.74401
-v 0.0545923 0.157542 0.773021
-v 0.0737315 0.11781 0.744577
-v 0.030002 0.117514 0.715566
-v 0.0807699 0.464601 0.530226
-v 0.0909607 0.481755 0.518118
-v 0.0946886 0.467032 0.500397
-v 0.0844977 0.449879 0.512505
-v 0.391726 0.221513 0.148065
-v 0.400795 0.21426 0.11948
-v 0.39973 0.244066 0.111579
-v 0.390661 0.251319 0.140163
-v 0.414426 0.280302 0.251288
-v 0.423294 0.281474 0.246648
-v 0.42106 0.291128 0.244816
-v 0.412192 0.289956 0.249455
-v 0.440455 0.284337 0.280096
-v 0.442584 0.285813 0.271643
-v 0.442667 0.294517 0.273184
-v 0.440538 0.293041 0.281637
-v 0.487749 0.272056 0.261181
-v 0.484524 0.275958 0.248756
-v 0.490299 0.287858 0.250995
-v 0.493524 0.283957 0.263419
-v 0.52765 0.24246 0.250057
-v 0.524761 0.246471 0.2314
-v 0.537916 0.260559 0.232393
-v 0.540805 0.256547 0.251049
-v 0.546745 0.216462 0.242839
-v 0.548035 0.216238 0.221776
-v 0.565217 0.228456 0.222698
-v 0.563927 0.22868 0.243761
-v 0.556432 0.199748 0.24957
-v 0.560467 0.196238 0.228355
-v 0.577406 0.210055 0.229291
-v 0.573371 0.213565 0.250507
-v 0.57224 0.183742 0.254002
-v 0.576896 0.179298 0.232565
-v 0.591174 0.196529 0.232095
-v 0.586517 0.200972 0.253532
-v 0.594314 0.169779 0.265262
-v 0.600191 0.164088 0.244486
-v 0.609043 0.184358 0.241438
-v 0.603167 0.190049 0.262213
-v 0.628873 0.164357 0.274639
-v 0.631799 0.159388 0.253769
-v 0.635884 0.180195 0.249389
-v 0.632958 0.185163 0.270259
-v 0.667251 0.157989 0.277862
-v 0.668156 0.153318 0.256368
-v 0.671928 0.174545 0.251914
-v 0.671022 0.179216 0.273408
-v 0.657917 0.245036 0.17911
-v 0.640608 0.242569 0.173637
-v 0.635062 0.255534 0.185332
-v 0.652371 0.258001 0.190805
+v -0.00938386 0.393382 0.51734
+v -0.0218465 0.415075 0.495837
+v 0.00847551 0.421064 0.484306
+v 0.0209381 0.399371 0.505808
+v 0.140344 0.240266 0.604737
+v 0.160612 0.263647 0.632013
+v 0.196166 0.255084 0.612934
+v 0.175898 0.231703 0.585658
+v 0.0478381 0.200333 0.674714
+v 0.0728928 0.219984 0.711491
+v 0.109845 0.189643 0.702528
+v 0.0847905 0.169992 0.665752
+v 0.00910727 0.143718 0.728895
+v 0.0361554 0.164933 0.768547
+v 0.075487 0.131339 0.759692
+v 0.0484389 0.110123 0.72004
+v 0.0812365 0.451773 0.521053
+v 0.0836402 0.473451 0.529359
+v 0.0942219 0.47986 0.50957
+v 0.0918183 0.458182 0.501264
+v 0.392153 0.220038 0.147159
+v 0.401105 0.215243 0.118024
+v 0.399303 0.245542 0.112484
+v 0.390351 0.250336 0.141619
+v 0.414835 0.280011 0.251178
+v 0.423519 0.281875 0.246418
+v 0.420651 0.291419 0.244925
+v 0.411967 0.289555 0.249685
+v 0.440505 0.284173 0.279858
+v 0.442635 0.286055 0.271487
+v 0.442618 0.294681 0.273422
+v 0.440488 0.292799 0.281793
+v 0.487584 0.271914 0.260908
+v 0.484574 0.276251 0.248574
+v 0.490464 0.288 0.251267
+v 0.493474 0.283664 0.263601
+v 0.527346 0.242273 0.249673
+v 0.524964 0.246822 0.231069
+v 0.53822 0.260746 0.232776
+v 0.540602 0.256196 0.25138
+v 0.546428 0.216213 0.242381
+v 0.548419 0.216488 0.221374
+v 0.565533 0.228705 0.223156
+v 0.563543 0.22843 0.244163
+v 0.556181 0.199403 0.249116
+v 0.560895 0.196453 0.227955
+v 0.577657 0.2104 0.229745
+v 0.572943 0.21335 0.250906
+v 0.572055 0.183314 0.253574
+v 0.577278 0.179562 0.232136
+v 0.591358 0.196957 0.232522
+v 0.586135 0.200708 0.253961
+v 0.594262 0.169278 0.264907
+v 0.600479 0.164382 0.244028
+v 0.609096 0.184859 0.241793
+v 0.602879 0.189754 0.262671
+v 0.628853 0.163844 0.274297
+v 0.631941 0.159718 0.253268
+v 0.635904 0.180707 0.249731
+v 0.632816 0.184834 0.27076
+v 0.667195 0.157474 0.277507
+v 0.668252 0.153663 0.255851
+v 0.671984 0.17506 0.252268
+v 0.670927 0.178872 0.273924
+v 0.658001 0.245148 0.179235
+v 0.640775 0.242493 0.173593
+v 0.634978 0.255422 0.185207
+v 0.652204 0.258077 0.190849
 f 9 13 14
 f 10 14 15
 f 11 15 16