set seed

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import re
-from utils import read_pcd, render_point_cloud, render_pcd_file
+from utils import read_pcd, render_point_cloud, render_pcd_file, set_seed
 from inference.utils import get_legend
 from inference.inference import segment_obj, get_heatmap
 from huggingface_hub import login
@@ -42,6 +42,7 @@ source_dict = {
 }
 
 def predict(pcd_path, inference_mode, part_queries):
+    set_seed()
     xyz, rgb, normal = read_pcd(pcd_path)
     if inference_mode == "Segmentation":
         parts = [part.strip(" ") for part in re.split(r'[,;.|]', part_queries)]
@@ -118,21 +119,6 @@ with gr.Blocks(theme=gr.themes.Default(text_size="lg", radius_size="none")) as d
             gallery_objaverse.select(fn=on_select, 
                            inputs=None, 
                            outputs=[file_upload, part_queries])
-            '''
-            gr.Examples(
-                inputs=[file_upload, part_queries],
-                examples=[
-                    ["examples/objaverse/fireplug.pcd", "bonnet of a fireplug,side cap of a fireplug,barrel of a fireplug,base of a fireplug"],
-                    ["examples/objaverse/mickey.pcd", "ear,head,arms,hands,body,legs"],
-                    ["examples/objaverse/motorvehicle.pcd", "wheel of a motor vehicle,seat of a motor vehicle,handle of a motor vehicle"],
-                    ["examples/objaverse/teddy.pcd", "head,body,arms,legs"],
-                    ["examples/objaverse/lamppost.pcd", "lighting of a lamppost,pole of a lamppost"],
-                    ["examples/objaverse/shirt.pcd", "sleeve of a shirt,collar of a shirt,body of a shirt"]
-                ],
-                example_labels=["fireplug", "Mickey", "motor vehicle", "teddy bear", "lamppost", "shirt"],
-                label=""
-            )
-            '''
         with gr.Column(scale=6):
             title = gr.HTML("""<h1 text-align="center">In the Wild</h1>
         <p style='font-size: 16px;'>Challenging in-the-wild reconstructions from iPhone photos & AI-generated images!</p>
@@ -147,20 +133,6 @@ with gr.Blocks(theme=gr.themes.Default(text_size="lg", radius_size="none")) as d
             gallery_wild.select(fn=on_select, 
                            inputs=None, 
                            outputs=[file_upload, part_queries])
-            '''
-            gr.Examples(
-                inputs=[file_upload, part_queries],
-                examples=[
-                    ["examples/wild/mcc_dalle_capybara.pcd", "hat worn by a capybara,head,body,feet"],
-                    ["examples/wild/mcc_dalle_corgi.pcd", "head,leg,body,ear"],
-                    ["examples/wild/mcc_iphone_pushcar.pcd", "wheel,body,handle"],
-                    ["examples/wild/mcc_iphone_plant.pcd", "pot,plant"],
-                    ["examples/wild/mcc_iphone_chair.pcd", "back of chair,leg,seat"],
-                ],
-                example_labels=["DALLE-capybara", "DALLE-corgi", "iPhone-pushcar", "iPhone-plant", "iPhone-chair"],
-                label=""
-            )
-            '''
 
     file_upload.change(
         fn=render_pcd_file,

inference/utils.py

@@ -39,13 +39,6 @@ def load_model():
     model = model.to(DEVICE)
     return model
 
-def set_seed(seed):
-    torch.manual_seed(seed)
-    if DEVICE != "cpu":
-        torch.cuda.manual_seed(seed)
-        torch.cuda.manual_seed_all(seed)
-    np.random.seed(seed)
-    random.seed(seed)
 
 def fnv_hash_vec(arr):
     """

utils.py

@@ -2,6 +2,18 @@ import plotly.graph_objects as go
 import open3d as o3d
 import numpy as np
 import textwrap
+import torch
+import random
+
+
+def set_seed():
+    seed = 123
+    torch.manual_seed(seed)
+    torch.cuda.manual_seed(seed)
+    torch.cuda.manual_seed_all(seed)
+    np.random.seed(seed)
+    random.seed(seed)
+
 
 def read_pcd(pcd_path):
     pcd = o3d.io.read_point_cloud(pcd_path)