chrockey commited on
Commit
02ecc6a
·
verified ·
1 Parent(s): 27599a2

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +6 -15
README.md CHANGED
@@ -33,26 +33,17 @@ Both subfolders contain a flat `config.json` + `model.safetensors`. SLAT + mesh/
33
 
34
  ## Usage
35
 
36
- Reconstruction:
37
-
38
- ```python
39
- from affostruction import ReconstructionPipeline
40
-
41
- pipeline = ReconstructionPipeline.from_pretrained("chrockey/Affostruction").cuda()
42
- outputs = pipeline.run(input_dict)
43
- mesh = outputs["mesh"][0]
44
- gaussian = outputs["gaussian"][0]
45
- ```
46
-
47
- Reconstruction + text-conditioned affordance heatmap:
48
-
49
  ```python
50
  from affostruction import AffostructionPipeline
51
 
52
  pipeline = AffostructionPipeline.from_pretrained("chrockey/Affostruction").cuda()
53
  outputs = pipeline.run(input_dict, queries=["grasp"])
 
 
54
  coords = outputs["coords"] # (N, 4) sparse voxel coords
55
- probs = outputs["affordance"][0]["probs"] # (N,) per-voxel heatmap in [0, 1]
56
  ```
57
 
58
- See `examples/reconstruction.py` and `examples/affordance.py` in the [GitHub repo](https://github.com/chrockey/Affostruction).
 
 
 
33
 
34
  ## Usage
35
 
 
 
 
 
 
 
 
 
 
 
 
 
 
36
  ```python
37
  from affostruction import AffostructionPipeline
38
 
39
  pipeline = AffostructionPipeline.from_pretrained("chrockey/Affostruction").cuda()
40
  outputs = pipeline.run(input_dict, queries=["grasp"])
41
+
42
+ mesh = outputs["mesh"][0]
43
  coords = outputs["coords"] # (N, 4) sparse voxel coords
44
+ probs = outputs["affordance"][0]["probs"] # (N,) per-voxel heatmap in [0, 1] (paired with coords)
45
  ```
46
 
47
+ > **Reconstruction only:** drop `queries` to skip affordance, or pass `load_affordance=False` to `from_pretrained` to avoid downloading the affordance denoiser.
48
+
49
+ See `examples/affordance.py` (full pipeline) and `examples/reconstruction.py` / `examples/reconstruction_unposed.py` (reconstruction-only) in the [GitHub repo](https://github.com/chrockey/Affostruction).