Hi,
I'm trying to create a custom regression output for my CNN. The basic premise is I take an image with various point-spread functions, and output an image with "spikes" at the center of those PSFs, and zero everywhere else.
I input and output images, so my loss function is an L2 loss function based off the different pixel values. To calculate this, I convolve the images with a 2D-Gaussian filter
L = ||T * g - Y * g||^2 + L1; where * denotes the convolution operator, T is the desired output, and Y is the output from the neural net.
My issue is that Y and T are dlarray objects, and imgaussfilt is incompatible with them. Is there a way around this?
Thanks!

답변 (1개)

Srivardhan Gadila
Srivardhan Gadila 2020년 8월 19일

0 개 추천

If you are looking for converting the Y and T from dlarray objects to single or double array then you can make use of the extractdata function.

카테고리

도움말 센터File Exchange에서 Custom Training Using Automatic Differentiation에 대해 자세히 알아보기

질문:

2020년 5월 20일

답변:

2020년 8월 19일

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by