|
Algorithm 1. Training algorithm. |
|
Require: = input images |
|
Require: L = set of training input indices with known labels |
|
Require: = labels for labeled inputs, i ∈ L
|
|
Require: = bounding box of sample i
|
|
Require:W_u = unsupervised loss weight |
|
Require: = neural network with trainable parameters as student model |
|
Require: = neural network as teacher model whose parameter with initial value
|
|
Require: = moving average momentum for parameters |
|
Require: = random Gaussian noise added to the input |
|
fort in [1, num_iterations] do
|
| draw a mini-batch B from labeled and unlabeled samples randomly |
|
evaluate network outputs |
|
evaluate network outputs |
| Find triplets T of components <i, j, k> in B where and is the pseudo label given by if
|
|
|
|
++
|
| update using ADAM optimizer |
| update by
|
| end for |
| return
|