chrockey commited on
Commit
21c5fc5
·
verified ·
1 Parent(s): 02ecc6a

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -39,11 +39,12 @@ from affostruction import AffostructionPipeline
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).
 
39
  pipeline = AffostructionPipeline.from_pretrained("chrockey/Affostruction").cuda()
40
  outputs = pipeline.run(input_dict, queries=["grasp"])
41
 
 
42
  coords = outputs["coords"] # (N, 4) sparse voxel coords
43
  probs = outputs["affordance"][0]["probs"] # (N,) per-voxel heatmap in [0, 1] (paired with coords)
44
  ```
45
 
46
+ > **Want a mesh / gaussian?** Pass `formats=["mesh", "gaussian"]` to `pipeline.run` to decode renderable outputs. Decoding is opt-in to avoid pulling in nvdiffrast / kaolin when callers only need the heatmap.
47
+ >
48
  > **Reconstruction only:** drop `queries` to skip affordance, or pass `load_affordance=False` to `from_pretrained` to avoid downloading the affordance denoiser.
49
 
50
  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).