search

ethz-asl / segmap

A map representation based on 3D segments
BSD 3-Clause "New" or "Revised" License
1.05k stars 394 forks source link

How to train the CNN with my own sensor information #152

Open david-villagra opened 4 years ago

david-villagra commented 4 years ago

Hi,

First of all, thank you for this great open-source SLAM. It looks amazingly promising with the current data-driven descriptors.

I have recently started using SegMap, and I was currently looking at how to train the CNN autoencoder and the semantic classifier, based on my own sensor information, in order to obtain my own CNN model and use it for my application (which is an indoor environment, so the segments look quite different).

I followed the tutorial on how to train a new CNN model and semantic classifier, but I could not find any documentation on how to generate your own dataset from your data (that would mean, obatining the .csv files required to actually train a CNN model).

I have seen that there are some tools for writing .csv files in the segmappy's script import_export.py, such as this function. But I haven't found any other script that uses those functions, or any related documentation.

The main question that I would like to ask would be:

  1. What is needed in order to generate all .csv files? It seems that the datasets can be generated by inputting the information to these write functions, such as the classes, matches, features..., or by inputting a .pcd file that contains the segments to be saved as a .csv file. Is there any additional information about the format of the input that these functions expect?

  2. How can I use these tools? Can I generate the dataset .csv files after having obtained a target map with all its segments using SegMap (with either the eigenvalue-based descriptor, or another descriptor)? Or how should be the correct way of obtaining these .csv files?
    I was wondering if you could help me figuring out covering how to actually use those functions and the format of the input that they expect. Is there maybe another script of tools for generating datasets that could be shared as well within SegMap?

Thanks in advance for the help!

ozzCb commented 4 years ago

Hi, I have the same question, seems you have get training data by running SegMap, does it work well?

Zedzou commented 3 years ago

hello, Did you have trained the CNN with your own senor data, I have the same question and are very confused. Thanks in advance for help.

cstlove commented 2 years ago

I had the same problem and was confused for a long time, did you solve this problem? I hope I can get your help!