Message Boards

WOLFRAM COMMUNITY

6151 Views

1 Reply

1 Total Likes

View groups...

Follow this post

Share this post:

GROUPS:

Avoid error in ImageAugmentationLayer usage?

Jeff Burns

Posted 7 years ago

This code fails with "NetTrain::interr: An internal error occurred. Please contact Wolfram Research. " Any ideas? lnet = NetChain[ { ImageAugmentationLayer[{115, 115}, "ReflectionProbabilities" -> {0.5, 0.5}], LinearLayer[1788], LogisticSigmoid, LinearLayer[1788], 2, SoftmaxLayer[]}, "Output" -> NetDecoder[{"Class", {1, 2}}], "Input" -> NetEncoder[{"Image", {125, 125}}] ] ; tSet = Table[ RandomChoice[{RegionImage[Annulus[], RasterSize -> 125] -> "A", RegionImage[Circle[], RasterSize -> 125] -> "C"}], 10] NetTrain[lnet, tSet]

This code fails with "NetTrain::interr: An internal error occurred. Please contact Wolfram Research. " Any ideas?

lnet = NetChain[
   {
    ImageAugmentationLayer[{115, 115}, 
     "ReflectionProbabilities" -> {0.5, 0.5}],
    LinearLayer[1788],
    LogisticSigmoid,
    LinearLayer[1788],
    2,
    SoftmaxLayer[]},
   "Output" -> NetDecoder[{"Class", {1, 2}}],
   "Input" -> NetEncoder[{"Image", {125, 125}}]
   ] ;

tSet = Table[
  RandomChoice[{RegionImage[Annulus[], RasterSize -> 125] -> "A", 
    RegionImage[Circle[], RasterSize -> 125] -> "C"}], 10] 

NetTrain[lnet, tSet]

POSTED BY: Jeff Burns