Stack Overflow
Stack Overflow

Questions tagged [nvidia-digits]

NVIDIA Deep Learning GPU Training System (DIGITS) is a GUI for training and developing deep neural networks.

See more on NVIDIA's site.
You can also help developing DIGITS on its github.

91 questions
1
vote
0 answers

Does the Caffe machine learning framework + Nvidia DIGITS software work with Python 3?

I installed Caffe and DIGITs on python 3 and it seems to work partially, however I still get this Protobuf error about version 2.6.1 versus 3.5. So I reinstalled Protobuf and was then following through with reinstalling Caffe. I have an Anaconda…
krishnab
  • 9,270
  • 12
  • 66
  • 123
1
vote
0 answers

Training FCN8s semantic segmentation on DIGITS/Caffe multi GPU

Has anyone had success in training a semantic segmentation model (FCN8s) using DIGITS (v5) in multi-GPU mode? I can successfully train on a single GPU, but it's very slow. When I select multiple GPUs (my workstation has 4 Titan Xp), the entire…
pepe
  • 9,799
  • 25
  • 110
  • 188
1
vote
1 answer

what is wrong when training an autoencoder on mnist dataset with caffe?

I want to use mnist dataset to train a simple autoencoder in caffe and with nvidia-digits. I have: caffe: 0.16.4 DIGITS: 5.1 python 2.7 I use the structure provided…
azzz
  • 91
  • 7
1
vote
0 answers

how to train an autoencoder for MNIST dataset in NVIDIA-Digits using Caffe?

I want to implement autoencoder on mnist_generic with caffe in NVIDIA digits. I made the dataset based on the approach explained here: https://github.com/NVIDIA/DIGITS/tree/master/examples/autoencoder But I do not want to use Torch or Tensorflow…
azzz
  • 91
  • 7
1
vote
1 answer

Define TensorFlow network key names according to an existing checkpoint

I trained a LeNet-gray-28x28 image detection Tensorflow model using Nvidia DIGITS, giving me the results I expect. Now, I have to classify some images outside of DIGITS and I want to use the model I trained to. So I get the LeNet model used by…
Damien Picard
  • 381
  • 4
  • 12
1
vote
0 answers

how to save visualization output of NVIDIA DIGITS to a folder?

I am using NVIDIA DIGITS for image classification. After I train my network and I test the model on the test data, I want to save the visualization and statistics that DIGITS can generate in my defined folder. How can I do it? For example how can I…
azzz
  • 91
  • 7
1
vote
0 answers

Nvidia DIGITS: Loss coverage (val and train) go straight to zero

EDIT: FOUND THE PROBLEM. I forgot to add a custom class when creating the dataset. Amateur mistake, but leaving this up for others who make a similar error. After successfully following the KITTI tutorial provided by NVIDIA, I began trying to run…
Allendorf
  • 21
  • 3
1
vote
1 answer

Nvidia digits object detection own dataset

According to information on the Nvidia website Digits uses datatasets in Kitti format. Is there possibilty in Digits or in external application to prepare such dataset or I will have to write it on my own? I would like to simply draw bounding boxes…
Mieszko
  • 69
  • 7
1
vote
1 answer

fcn_alexnet.caffemodel is not visible in standard segmentation network of DIGITS

I wanted to use NVIDIA DIGITS for image segmentation. According to this tutorial, https://github.com/NVIDIA/DIGITS/tree/master/examples/semantic-segmentation , I have created caffmodel named as 'fcn_alexnet.caffemodel'. But, it is not showing on my…
Gunjan naik
  • 501
  • 1
  • 7
  • 18
1
vote
2 answers

Nvidia Digits accuracy and loss plots data

I trained my model in Nvidia Digits 5 and I would now like to extract the accuracy and loss plots that were generated during training for a report. Is this data saved somewhere so that it would possible to extract the data for these plots so that I…
Johann
  • 73
  • 8
1
vote
0 answers

NVIDIA DIGITS: Get accuracy for segmented images

I am trying to use NVIDIA DIGITS to perform image segmentation (by following this article). After training the model, I was able to perform the segmentation on individual images. Now I would like to give a list of test images and labeled images to…
user2312896
  • 389
  • 3
  • 14
1
vote
1 answer

How can I import external prebuilt .LMDB files into NVIDIA DIGITS?

I have several databases and I need to do classification on them on NVIDIA DIGITS. But importing my big data into DIGITS takes a lot of time ( 2-4 days)!!! Imagine I have converted 2 image sets into .lmdb forms like: data1 …
azzz
  • 91
  • 7
1
vote
0 answers

Crash while training Caffe for object detection

Training using DetectNET model provided by Kitti on Tesla-P40(22.4 GB) GPU. F0412 12:27:08.371733 26664 detectnet_transform_layer.cpp:212] Check failed: bboxLen == sizeof(BboxLabel) / sizeof(Dtype) (237 vs. 16) *** Check failure stack trace:…
danishansari
  • 634
  • 5
  • 21
1
vote
1 answer

How to apply a pre-trained model of 3 channel images on single channel images?

I tried to used a pre-trained model that already was trained on three-channel color images, however, I am getting an error because of shape difference. Could someone let me know how can I tackle this issue? One user had suggested using Tile layer,…
S.EB
  • 1,966
  • 4
  • 29
  • 54
1
vote
1 answer

ImportError: No module named skfmm

I am trying to install "DIGITS" in a virtual environment, I am receiving the following error: digits/webapp.py:7: ExtDeprecationWarning: Importing flask.ext.socketio is deprecated, use flask_socketio instead. from flask.ext.socketio import…
S.EB
  • 1,966
  • 4
  • 29
  • 54
1 2
3
4 5 6 7