Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support for inference on LMDB #638

Merged
merged 1 commit into from
Apr 10, 2016
Merged

Conversation

gheinrich
Copy link
Contributor

A (small) step towards support for non-image data: this makes it possible to perform inference on any 3D blob. Blobs may not be "proper" images (e.g. because they don't have a standard number of channels).

close #619
close #630

A new field and a new button were added to allow inference on LMDB database:

generic-inference-form

Results are shown as follows (using DB key instead of image path):

infer_db

Progress

  • Add "Test DB" form to generic model show page
  • Create new route
  • Update inference job to take in a path to a database
  • Update inference tool to parse an LMDB database
  • Add tests

@gheinrich
Copy link
Contributor Author

Tests added in latest commit.

@gheinrich gheinrich merged commit 2b1cf20 into NVIDIA:master Apr 10, 2016
@gheinrich gheinrich deleted the dev/lmdb-inference branch April 10, 2016 08:57
@ajcampisi
Copy link

ajcampisi commented May 22, 2016

I am interested in using Digits with my Multispectral images (4 image channels). Since most multi-spectral images can be output as a separate 8 bit gray-scale images (one for each channel), I was wondering if we could get Digits to accept each multi-spectral image as four separate channel images for both training and classification. For example the file-list could contain the image name, channel number and category.
Image1.jpg c1 1
Image1.jpg c2 1
Image1.jpg c3 1
Image1.jpg c4 1
Image2.jpg c1 1
Image2.jpg c2 1
Image2.jpg c3 1
Image2.jpg c4 1
Image3.jpg c1 2
Image3.jpg c2 2
Image3.jpg c3 2
Image3.jpg c4 2

@gheinrich
Copy link
Contributor Author

Hi @ajcampisi DIGITS does not support the format you propose out of the box. Are you saying that each image channel - taken in isolation - is sufficient to classify the image? In that case why don't you convert the 4-channel image to grayscale with:

convert <input> -colorspace Gray <output>

and then feed those grayscale images to DIGITS?

@ajcampisi
Copy link

Hi Greg,
Thanks for the reply. No unfortunately it is not sufficient an individual channel image to be used for classification. All four channels images need to be feed into the network as four separate network inputs and the combined image data from all four network channels can then be used by the network for both classification and training.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Optionally keep the information about the original image in LMDB Support for inference on LMDB file
2 participants