search

NVIDIA / DIGITS

Deep Learning GPU Training System
https://developer.nvidia.com/digits
BSD 3-Clause "New" or "Revised" License
4.12k stars 1.38k forks source link

Q: Running the segmentation outside DIGITS (python files which contribute to make the visual segmentation) #1614

Open MyVanitar opened 7 years ago

MyVanitar commented 7 years ago

Hello,

I've trained a segmentation model using DIGITS, but I want to run the trained segmentation model outside DIGITS. Which python files should be used for this and they are in which DIGITS folder?

BR

MyVanitar commented 7 years ago

@gheinrich

can you help Greg?

gheinrich commented 7 years ago

You can inspect the code in https://github.com/NVIDIA/DIGITS/tree/master/digits/extensions/view/imageSegmentation to see how we are rendering the output of a segmentation model in DIGITS.

MyVanitar commented 7 years ago

@gheinrich

Thanks. So there is no single python file (like classification task) to get the trained model as input and handle the results, yes?

gheinrich commented 7 years ago

You can reuse the classification example almost entirely if you're fine with bounding box coordinates. I thought you were asking for the actual visualization of segmented areas.

MyVanitar commented 7 years ago

@gheinrich

Actually you are right, I'm looking for the visualization of the segmented area. I thought view.py will do the same task as classify.py does.

MyVanitar commented 7 years ago

@gheinrich @jmancewicz

I investigated the code a bit, but I would appreciate if you just let me where should I import the trained .caffemodel or other necessary files in the code?

it will save a lot of time for me.

MyVanitar commented 7 years ago

@gheinrich

reminder

gheinrich commented 7 years ago

Sorry I don't have time to help there.