Skip to content

Commit

Permalink
Update inference.md
Browse files Browse the repository at this point in the history
Remove mention to downloading dataset when not needed
  • Loading branch information
MedericFourmy authored Mar 25, 2024
1 parent 0720cce commit 294b529
Showing 1 changed file with 4 additions and 11 deletions.
15 changes: 4 additions & 11 deletions docs/book/cosypose/inference.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# Inference

Here are provided the minimal commands you have to run in order to run the inference of CosyPose. You need to set up the environment variable `$HAPPYPOSE_DATA_DIR` as explained in the README.
Here are provided the minimal commands you have to run in order to run the inference of CosyPose on the barbecu-sauce example. You need to set up the environment variable `$HAPPYPOSE_DATA_DIR` as explained in the README.

## 1. Download pre-trained pose estimation models

Expand All @@ -11,26 +11,19 @@ python -m happypose.toolbox.utils.download --cosypose_models \
coarse-bop-hope-pbr--225203 \
refiner-bop-hope-pbr--955392
```

## 2. Download YCB-V Dataset

```sh
python -m happypose.toolbox.utils.download --bop_dataset=ycbv
```

## 3. Download the example
## 2. Download the example data

```sh
python -m happypose.toolbox.utils.download --examples barbecue-sauce
```

## 4. Run the script
## 3. Run the script
The example contains default outputs for detection and pose prediction
```sh
python -m happypose.pose_estimators.cosypose.cosypose.scripts.run_inference_on_example barbecue-sauce --run-inference --run-detections --vis-detections --vis-poses
```

## 5. Results
## 4. Results

The results are stored in the visualization folder created in the example directory.

Expand Down

0 comments on commit 294b529

Please sign in to comment.