Skip to main content
. 2021 Feb 18;7:e358. doi: 10.7717/peerj-cs.358

Algorithm 2. Build a deep Learning model using CXRVN-proposed architecture.

Input Image_COVID-19_Set imds, Generator Layers GeLayers, Discriminator Layers DiLayers
Output Generator GEN, Discriminator DISC
  1. Begin

  2.  // Setting up Training options

  3.  Options.set(ValidationFrequecy 5)

  4.  Options.set(InitialLearnRate ←1e-4)

  5.  Options.set(LearnRateSchedule ← Piecewise)

  6.  Options.set(MiniBatchSize ←16)

  7.  Options.set(MaxEpochs ←50)

  8.  GAN LGraph2Net(GeLayers)

  9.  DISC LGraph2Net(DiLayers)

  10.  // Training GEN and DISC using imds

  11.  For i=1: MaxEpochs

  12.   batchImgs read(imds,BatchSize)

  13.   Imgs Shulfe(batchImgs)

  14.   latentIN gen(Imgs,GEN)

  15.   DPred Forward(DISC, latentIN)

  16.   GPred Forward(GEN, latentIN)

  17.   DProb Sigmoid(DPred)

  18.   GProb Mean(DProb)

  19.   Loss CalcLoss(DProb, GProb)

  20.   GAN CalcGradients(GAN.Learnables,Loss)

  21.   DISC CalcGradients(DISC.Learnables,Loss)

  22.  End for

  23. End