geninhu commited on
Commit
d346b49
·
1 Parent(s): 230767f

Update application file

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -68,6 +68,17 @@ def tnet(inputs, num_features):
68
  EXAMPLES_PATH = './examples'
69
  model = from_pretrained_keras('keras-io/PointNet')
70
 
 
 
 
 
 
 
 
 
 
 
 
71
  def infer(img_path):
72
  mesh = trimesh.load(img_path.name)
73
  points = mesh.sample(2048)
 
68
  EXAMPLES_PATH = './examples'
69
  model = from_pretrained_keras('keras-io/PointNet')
70
 
71
+ CLASS_MAP = {0: 'chair',
72
+ 1: 'sofa',
73
+ 2: 'desk',
74
+ 3: 'bed',
75
+ 4: 'dresser',
76
+ 5: 'night_stand',
77
+ 6: 'toilet',
78
+ 7: 'bathtub',
79
+ 8: 'monitor',
80
+ 9: 'table'}
81
+
82
  def infer(img_path):
83
  mesh = trimesh.load(img_path.name)
84
  points = mesh.sample(2048)