Diff-UNet
Diff-UNet copied to clipboard
BTCV/test.py label input.
Program file "BTCV/text.py " a line of code at the line number 100 below. This line of code should be batch["label"] instead of batch["raw_label"]. Because batch["raw_label"] does not match with input's volume sizes.
label = batch["raw_label"] --> label = batch["label"]
You should use the "raw_label" in the test phase.
We should compute the metric using the "out" and "label" with original size.