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