f.write(
";%s" %
list_to_string(a['plans_per_stage'][stages[stage]]['current_spacing']))
f.write(";%s" % list_to_string(
a['plans_per_stage'][stages[stage]]['original_spacing']))
f.write(";%s" %
str(a['plans_per_stage'][stages[stage]]['pool_op_kernel_sizes']))
f.write(";%s" %
str(a['plans_per_stage'][stages[stage]]['conv_kernel_sizes']))
if do_linebreak_at_end:
f.write("\n")
if __name__ == "__main__":
summarize((1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 24, 27),
output_dir=join(network_training_output_dir, "summary_fold0"),
folds=(0, ))
base_dir = os.environ['RESULTS_FOLDER']
nnunets = ['nnUNetV2', 'nnUNetV2_zspacing']
task_ids = list(range(99))
with open("summary.csv", 'w') as f:
f.write(
"identifier;stage;batch_size;num_pool_per_axis;patch_size;patch_size(mm);median_patient_size_in_voxels;median_patient_size_in_mm;current_spacing;original_spacing;pool_op_kernel_sizes;conv_kernel_sizes;patient_dc;global_dc\n"
)
for i in task_ids:
for nnunet in nnunets:
try:
summary_folder = join(base_dir, nnunet, "summary_fold0")
if isdir(summary_folder):
summary_files = subfiles(summary_folder,
join=False,
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from nnunet.evaluation.model_selection.summarize_results_in_one_json import summarize
from nnunet.paths import network_training_output_dir
from batchgenerators.utilities.file_and_folder_operations import *
if __name__ == "__main__":
summary_output_folder = join(network_training_output_dir, "summary_jsons_fold0")
maybe_mkdir_p(summary_output_folder)
summarize(range(50), output_dir=summary_output_folder, folds=(0,))
results_csv = join(network_training_output_dir, "summary_fold0.csv")
summary_files = subfiles(summary_output_folder, suffix='.json', join=False)
with open(results_csv, 'w') as f:
for s in summary_files:
if s.find("ensemble") == -1:
task, network, trainer, plans, validation_folder = s.split("__")
else:
n1, n2 = s.split("--")
n1 = n1[n1.find("ensemble_") + len("ensemble_") :]
task = s.split("__")[0]
network = "ensemble"
trainer = n1