Skip to main content
. 2020 Oct 19;58(1):102411. doi: 10.1016/j.ipm.2020.102411

Algorithm 4.

Pseudocode of CGNet framework.

Loading CNNs pre-trained on ImageNet;
Remove top layers and add new top layers for the classification task;
Train the modified CNNs on the training dataset;
Obtain features through a fully-connected layer whose output size is 1000 from the trained CNNs;
Divide features into batches of same sizes;
Build graphs in each batch of features;
Reconstruct features according to graph information;
Train neural networks with reconstructed features;
Acquire features for test set by repeat Step 4 to Step 7 on the test set;
Test the trained neural networks with the reconstructed test features.