search

Simple-Robotics / cosypose

Code for "CosyPose: Consistent multi-view multi-object 6D pose estimation", ECCV 2020.
MIT License
69 stars 15 forks source link

BOP inference on individual datasets #14

Closed swtyree closed 1 year ago

swtyree commented 1 year ago

One last PR for today: I added support for a --dataset flag in the BOP inference script (cosypose.scripts.run_bop_inference) to allow inference on a single dataset at a time. When used as below, the submission file can be generated for a single dataset:

python -m cosypose.scripts.run_bop_inference --config bop-pbr --dataset DATASET
python -m cosypose.scripts.run_bop20_eval_multi --result_id=RESULT_ID \
  --method=maskrcnn_detections/refiner/iteration=4 --convert_only
ylabbe commented 1 year ago

Looks good and useful to me. Thanks !