Thanks for your kind sharing.
I doubt that the 2 parts below are mixed up.
|
cv2.imwrite(os.path.join(preds_dir, |
|
"test_localization_" + sample["img_name"][i] + "_prediction.png"), |
|
mask.cpu().numpy()[i, 4]) |
|
cv2.imwrite(os.path.join(targs_dir, "test_localization_" + sample["img_name"][i] + "_target.png"), |
|
1 * (argmax > 0)) |
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
xview2_solution/train.py
Lines 294 to 295 in 5d0caba