search

KaiChen1998 / GeoDiffusion

Official PyTorch implementation of GeoDiffusion in ICLR 2024 (https://arxiv.org/abs/2306.04607)
https://kaichen1998.github.io/projects/geodiffusion/
MIT License
64 stars 3 forks source link

Training on custom dataset (KITTI) #12

Closed joonjeon closed 4 months ago

joonjeon commented 5 months ago

I am trying to use your work to train on KITTI dataset, but it does not work.

Error message says that num_samples becomes zero.

The following is my directory tree for the KITTI dataset: kitti-tree.txt

Any ideas on why this is happening?

KaiChen1998 commented 4 months ago

@joonjeon It seems like you might not construct the COCO-style annotation file correctly, and therefore, resulting in num_samples=0 even if there are indeed images saved in your directory tree. I do recommend you to check here to get familiar with the COCO-format, here to check the detailed image filtering procedure of COCO dataset, and here to get more familiar with the KITTI official annotation format.

joonjeon commented 4 months ago

Alright. Thanks a lot for your help.