use AVOD in python 2

[kargarisaac]

Hi,

I want to use AVOD with ROS1 and python2. I followed the steps in the readme file without any problem, but when I want to run " python scripts/preprocessing/gen_mini_batches.py" in avod folder, I get the following error:

File "bev_generator.py", line 16, in <module> from avod.builders.dataset_builder import DatasetBuilder File "/home/isaac/catkin_ws/src/bev/avod/avod/builders/dataset_builder.py", line 14, in <module> from avod.protos import kitti_dataset_pb2 ImportError: No module named protos the import part of dataset_builder.py is as follows:

import avod
from avod.datasets.kitti.kitti_dataset import KittiDataset
from avod.protos import kitti_dataset_pb2
from` avod.protos.kitti_dataset_pb2 import KittiDatasetConfig

I wonder why it knows avod and avod.datasets.kitti.kitti_dataset folders but doesn't know avod.protos.

Any suggestion?

Thank you

[ChunML]

Add a blank __init__.py inside protos folder (under avod folder) may do the trick.

[kargarisaac]

Add a blank __init__.py inside protos folder (under avod folder) may do the trick.

Thank you. It worked!

[sgs678]

Will you still need to install ROS to run this code? I will be very grateful for your answer.

[ChunML]

You don't have to install ROS. In fact, the source code is written in Python 3.5, so it won't work with ROS. If you want to use the code with ROS, you have to make the source compatible with Python 2 first (which can be done easily) and make a ROS node for that.

[sgs678]

That's good, I hope so. Thank you for your answer.

[astronaut71]

hi. I would like to use this with ROS too. And is it possible to use the code with latest CUDA 10 and cuddn 7.1? Also this this code can give me the individual orientation in real time of the detected objects?