首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >CUDA错误-语义分割

CUDA错误-语义分割
EN

Stack Overflow用户
提问于 2017-10-06 21:18:41
回答 2查看 179关注 0票数 0

使用语义分割,我想把卫星图像分成两类:waterland。我有这样的问题:

在CUDA执行过程中发生了意外错误。数据自动化系统的错误是: CUDA_ERROR_LAUNCH_FAILED

这是我的代码:

代码语言:javascript
复制
clear;clc;close all

dataDir = fullfile('C:\Users\firat\Desktop\TEZ\Uygulama\Semantic 
Segmentation\data');
imDir = fullfile(dataDir,'image');
pxDir = fullfile(dataDir,'imagePixelLabels');

imds = imageDatastore(imDir);

I = readimage(imds,1);
figure
imshow(I)

% imageLabeler(imDir);

classNames = ["Water" "Land"];
pixelLabelID = [1 2];
pxds = pixelLabelDatastore(pxDir,classNames,pixelLabelID);

C = readimage(pxds,1);

B = labeloverlay(I,C);
figure
imshow(B)

buildingMask = C == 'Water';
figure
imshowpair(I, buildingMask,'montage')

% Create a Semantic Segmentation Network

numFilters = 64;
filterSize = 3;
numClasses = 2;
layers = [
imageInputLayer([1024 1024 3])
convolution2dLayer(filterSize,numFilters,'Padding',1)
reluLayer()
maxPooling2dLayer(2,'Stride',2)
convolution2dLayer(filterSize,numFilters,'Padding',1)
reluLayer()
transposedConv2dLayer(4,numFilters,'Stride',2,'Cropping',1);
convolution2dLayer(1,numClasses);
softmaxLayer()
pixelClassificationLayer()
]

opts = trainingOptions('sgdm', ...
'InitialLearnRate', 1e-3, ...
'MaxEpochs', 100, ...
'MiniBatchSize', 64);

trainingData = pixelLabelImageSource(imds,pxds);

net = trainNetwork(trainingData,layers,opts);

testImage = imread('C:\Users\firat\Desktop\TEZ\Uygulama\Semantic 
Segmentation\test\test3.tif');

C = semanticseg(testImage,net);
B = labeloverlay(testImage,C);
figure
imshow(B)

我该如何解决这个问题?

EN

回答 2

Stack Overflow用户

发布于 2017-10-09 19:34:33

看起来像是某种数据自动化系统的设置问题。以下各项的产出:

代码语言:javascript
复制
gpuDevice()
票数 0
EN

Stack Overflow用户

发布于 2017-10-09 19:52:02

gpuDevice()

代码语言:javascript
复制
CUDADevice with properties:
                  Name: 'GeForce 840M'
                 Index: 1.00
     ComputeCapability: '5.0'
        SupportsDouble: 1
         DriverVersion: 9.00
        ToolkitVersion: 8.00
    MaxThreadsPerBlock: 1024.00
      MaxShmemPerBlock: 49152.00
    MaxThreadBlockSize: [1024.00 1024.00 64.00]
           MaxGridSize: [2147483647.00 65535.00 65535.00]
             SIMDWidth: 32.00
           TotalMemory: 2147483648.00
   MultiprocessorCount: 3.00
          ClockRateKHz: 1124000.00
           ComputeMode: 'Default'
  GPUOverlapsTransfers: 1
KernelExecutionTimeout: 1
      CanMapHostMemory: 1
       DeviceSupported: 1
        DeviceSelected: 1
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/46614049

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档