Skip to content

Mix up prediction and target in validate function #13

Description

@rrryan2016

Thanks for your kind sharing.

I doubt that the 2 parts below are mixed up.

xview2_solution/train.py

Lines 289 to 291 in 5d0caba

cv2.imwrite(os.path.join(preds_dir,
"test_localization_" + sample["img_name"][i] + "_prediction.png"),
mask.cpu().numpy()[i, 4])

xview2_solution/train.py

Lines 294 to 295 in 5d0caba

cv2.imwrite(os.path.join(targs_dir, "test_localization_" + sample["img_name"][i] + "_target.png"),
1 * (argmax > 0))

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions