Second stage (Source-Free Domain Adaptation) model loading BUG fixed

[zyfone]

stage

python tools/train_net_mcd.py --config ./configs/sf/dut_hcmus.yaml OUTPUT_DIR outputs/dut_hcmus

issue

valid_map = scores_pre >= max(scores_pre) ValueError: max() arg is an empty sequence

cause

TEACHER MODEL output is None

From function threshold_bbox

1）scores_pre:tensor([], device='cuda:0', grad_fn=<IndexBackward>)

2）list_of_boxes[BoxList(num_boxes=0, image_width=599, image_height=512, mode=xyxy), BoxList(num_boxes=0, image_width=640, image_height=512, mode=xyxy)]

print(checkpoint["model"].get("generator_t")) is None

from fcos_core/utils/checkpoint.py

Code modification solution

how to fix this bug

# elif checkpoint["model"].get("model_backbone"):
elif checkpoint["model"].get("generator_s") and not checkpoint["model"].get("generator_t"):
    self.logger.info("Loading the source only model v2 ==> ")
    load_state_dict(self.model["generator_t"], checkpoint["model"]["generator_s"])
    load_state_dict(self.model["generator_s"], checkpoint["model"]["generator_s"])
    load_state_dict(self.model["predictor_t"], checkpoint["model"]["predictor_s"])
    load_state_dict(self.model["predictor_s"], checkpoint["model"]["predictor_s"])