search

fvisin / reseg

A Recurrent Neural Network for Object Segmentation
GNU General Public License v3.0
125 stars 35 forks source link

my own dataset? #6

Closed SteveIb closed 6 years ago

SteveIb commented 6 years ago

Hi, is it possible to run your code on my own dataset? Regards.

fvisin commented 6 years ago

Of course, but you will need to modify the code a bit.

The easiest thing will be to create your own dataset mimicking camvid.py. You will just need to implement the properties function and the load_data one: https://github.com/fvisin/reseg/blob/8040d190fcdb6d191faf9e33abe671a9c2fa2dad/camvid.py#L274

These are the set of arguments that will be passed to the load_data function (note that you can ignore some of them if you so wish): https://github.com/fvisin/reseg/blob/8040d190fcdb6d191faf9e33abe671a9c2fa2dad/reseg.py#L615-L622

The last step should be to import your dataset and add it to the dict here: https://github.com/fvisin/reseg/blob/8040d190fcdb6d191faf9e33abe671a9c2fa2dad/reseg.py#L36

I cannot provide more support than this, but you should have all the information you need. Cheers!

SteveIb commented 6 years ago

Thanks.. I will try run in the coming days. Regards.