junjun-jiang / BaMBNet

The official repository for BaMBNet
22 stars 2 forks source link

Input for model in test.py #3

Closed yarqian closed 3 years ago

yarqian commented 3 years ago

Many thanks for your work. I am confused about the input for model in test.py. It seems that the COC maps are not used as input for model in test.py. While the COC maps ARE used as input for model in train.py. Is it used for ablation study or a wrong setting. Many thanks.

junjun-jiang commented 3 years ago

During the early training phase, the COC maps also required to be used as the inputs, and will be discarded after the network converges. As a consequence, the model does not require the COC maps during the test phase.