tikaiz/DSAI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed evaluation error

Browse Source
This commit is contained in:
Tim Kainz
2026-01-24 15:02:07 +01:00
parent 09d1911feb
commit 3b1d3c0497
7 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
image-inpainting/results/best_model.pt
View File

Binary file not shown.

BIN
image-inpainting/src/__pycache__/architecture.cpython-313.pyc Normal file
View File

Binary file not shown.

BIN
image-inpainting/src/__pycache__/datasets.cpython-313.pyc Normal file
View File

Binary file not shown.

BIN
image-inpainting/src/__pycache__/train.cpython-313.pyc Normal file
View File

Binary file not shown.

BIN
image-inpainting/src/__pycache__/utils.cpython-313.pyc Normal file
View File

Binary file not shown.

2
image-inpainting/src/main.py
View File

@@ -32,7 +32,7 @@ if __name__ == '__main__':
config_dict['print_train_stats_at'] = 10
config_dict['print_stats_at'] = 100
config_dict['plot_at'] = 10
config_dict['plot_at'] = 100
config_dict['validate_at'] = 100
network_config = {

2
image-inpainting/src/utils.py
View File

@@ -113,7 +113,7 @@ def create_predictions(model_config, state_dict_path, testset_path, device, save
print(f"Processing image {i + 1}/{len(input_arrays)}")
input_array = torch.from_numpy(input_arrays[i]).to(
device)
output = model(input_array)
output = model(input_array.unsqueeze(0) if hasattr(input_array, 'dim') and input_array.dim() == 3 else input_array)
output = output.cpu().numpy()
predictions.append(output)