Fix minor problems

This commit is contained in:
Emilio Martínez 2023-08-17 23:26:50 -03:00
parent 81d69cd191
commit a4150b46c1
3 changed files with 8 additions and 4 deletions

View File

@ -9,7 +9,7 @@
, "path": {
"root": "drunet_inference" // "denoising" | "superresolution"
, "pretrained_netG": "denoising/drunet/models/520_G.pth" // path of pretrained model, if model from scratch type: null
, "pretrained_netG": "denoising/drunet/models/50_G.pth" // path of pretrained model, if model from scratch type: null
}
, "datasets": {

View File

@ -4,7 +4,7 @@ category: '[Tempest]'
templates:
imports: import tempest
make: tempest.buttonToFileSink(${Filename}, ${input_width}, ${H_size}, ${V_size}, ${enhance_image})
make: tempest.buttonToFileSink(${Filename}, ${input_width}, ${H_size}, ${V_size}, ${remove_blanking}, ${enhance_image})
# Make one 'parameters' list entry for every parameter you want settable from the GUI.
# Keys include:

View File

@ -583,8 +583,12 @@ def apply_blanking_shift(I, h_active=1600, v_active=900,
plt.show()
# Angle and rho arrays
lines_angles = lines[:,0,1]
lines_rhos = lines[:,0,0]
try:
lines_angles = lines[:,0,1]
except:
print("No lines detected. Returning original image")
return I
# Find unique lines angles detected
unique_angles = np.unique(lines_angles)