shanice-l / gdrnpp_bop2022

PyTorch Implementation of GDRNPP, winner (most of the awards) of the BOP Challenge 2022 at ECCV'22
Apache License 2.0
229 stars 49 forks source link

training procedure/code for a custom dataset made by BlenderProc that follows linemod format #106

Closed monajalal closed 7 months ago

monajalal commented 8 months ago

I have created a synthetic dataset that follows lm format and its of a custom ply model (non-existent in BOP), how can I feed it to gdrnpp seemlessly? I checked the issues but cannot find the code for custom training of a data that follows lm format. Could you please provide some instructions?

(blenderproc) mona@ada:~/BlenderProc/examples/datasets/bop_challenge/output_data/bop_data$ ls lm
total 16K
drwxrwxr-x 3 mona mona 4.0K Jan  9 13:13 train_pbr
drwxrwxr-x 3 mona mona 4.0K Jan  9 13:13 ..
drwxrwxr-x 3 mona mona 4.0K Jan  9 13:13 .
-rw-rw-r-- 1 mona mona  166 Jan  9 13:29 camera.json

(blenderproc) mona@ada:~/BlenderProc/examples/datasets/bop_challenge/output_data/bop_data$ ls lm/train_pbr/000000/
total 2.4M
drwxrwxr-x 3 mona mona 4.0K Jan  9 13:13 ..
drwxrwxr-x 6 mona mona 4.0K Jan  9 13:14 .
drwxrwxr-x 2 mona mona  12K Jan  9 13:29 rgb
drwxrwxr-x 2 mona mona  12K Jan  9 13:29 depth
drwxrwxr-x 2 mona mona  12K Jan  9 13:29 mask
drwxrwxr-x 2 mona mona  12K Jan  9 13:29 mask_visib
-rw-rw-r-- 1 mona mona  80K Jan  9 13:29 scene_gt.json
-rw-rw-r-- 1 mona mona 111K Jan  9 13:29 scene_camera.json
-rw-rw-r-- 1 mona mona  51K Jan  9 13:29 scene_gt_info.json
-rw-rw-r-- 1 mona mona 2.1M Jan  9 13:29 scene_gt_coco.json
monajalal commented 8 months ago

@wangg12 could you please help with this? I couldn't find any instructions for these steps in issues or your README.md

https://github.com/shanice-l/gdrnpp_bop2022/issues/28

wangg12 commented 8 months ago

https://github.com/THU-DA-6D-Pose-Group/GDR-Net/discussions/62#discussioncomment-1715072