randing2000 commited on
Commit
fb54695
·
verified ·
1 Parent(s): dbaf344

Add simple TARGO-Net model card

Browse files
Files changed (1) hide show
  1. README.md +67 -0
README.md CHANGED
@@ -1,3 +1,70 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ language:
4
+ - en
5
+ tags:
6
+ - robotics
7
+ - grasping
8
+ - 3d-vision
9
+ - shape-completion
10
+ - pytorch
11
+ library_name: pytorch
12
  ---
13
+
14
+ # TARGO-Net
15
+
16
+ TARGO-Net is a PyTorch checkpoint release for target-oriented robotic grasping in cluttered scenes. This repository currently provides the model weights needed by the cleaned TARGO inference and training code.
17
+
18
+ ## Files
19
+
20
+ | File | Description |
21
+ | --- | --- |
22
+ | `checkpoints/targonet.pt` | TARGO grasp prediction network checkpoint. |
23
+ | `checkpoints/adapointr.pth` | AdaPoinTr shape completion checkpoint used by the TARGO inference pipeline. |
24
+
25
+ ## Usage
26
+
27
+ Download the checkpoints with `huggingface_hub`:
28
+
29
+ ```python
30
+ from huggingface_hub import snapshot_download
31
+
32
+ snapshot_download(
33
+ repo_id="randing2000/TARGO-Net",
34
+ local_dir="checkpoints_hf",
35
+ allow_patterns=["checkpoints/targonet.pt", "checkpoints/adapointr.pth"],
36
+ )
37
+ ```
38
+
39
+ For the cleaned TARGO code, place or link the downloaded files as:
40
+
41
+ ```text
42
+ checkpoints/targonet.pt
43
+ checkpoints/adapointr.pth
44
+ ```
45
+
46
+ Then run inference with the processed VGN-format test data:
47
+
48
+ ```bash
49
+ python inference_targo.py --model checkpoints/targonet.pt --sc_model_path checkpoints/adapointr.pth --test_root /path/to/processed_vgn/test_set_gaussian_0.002
50
+ ```
51
+
52
+ The `--test_root` directory should contain:
53
+
54
+ ```text
55
+ scenes/
56
+ mesh_pose_dict/
57
+ occ_level_dict.json
58
+ ```
59
+
60
+ ## Training / Evaluation Data
61
+
62
+ The checkpoints are intended for the TARGO/VGN-style data pipeline. The processed dataset is not included in this model repository.
63
+
64
+ ## Limitations
65
+
66
+ These weights are research artifacts and are intended for offline robotics research workflows. Real-robot deployment requires additional calibration, safety checks, and environment-specific validation.
67
+
68
+ ## License
69
+
70
+ MIT.