second.pytorch icon indicating copy to clipboard operation
second.pytorch copied to clipboard

Cannot prep kitti dataset or train

Open HukamiEstia opened this issue 5 years ago • 1 comments

Hi,

Ubuntu 18.04 python3.8 miniconda cuda 10.2 pytorch 1.5.1

I followed the instructions to install on the master branch (and spconv as well) but when running

python create_data.py kitti_data_prep /path/to/KITTI_DATASET_ROOT

I get a seg fault.

So I tried to find out where it happens, and I found that it is caused by import of function from second.core.non_max_suppression.nms_gpu

In the end I found that removing b[0] from the function trangle_area (misssing "i"?) allows the import and the script runs.

So I changed https://github.com/traveller59/second.pytorch/blob/3aba19c9688274f75ebb5e576f65cfe54773c021/second/core/non_max_suppression/nms_gpu.py#L169

to (a[0] - c[0]) * (b[1] - c[1]) - (a[1] - c[1])) / 2.0 and it runs successfully. However I have no idea what effect it has on the output...

Since I'm not sure it is even used I tried to train with:

python ./pytorch/train.py train --config_path=./configs/car.fhd.config --model_dir=/path/to/model_dir

It looks like it start but at some point I get another segmentation fault... and warnings from numba.

HukamiEstia avatar Jul 24 '20 15:07 HukamiEstia

@HukamiEstia I am facing the same issue of multiple segfaults. Did you get any solution to this?

patilunmesh avatar Dec 11 '20 15:12 patilunmesh