Fix minor problems
This commit is contained in:
parent
81d69cd191
commit
a4150b46c1
|
@ -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": {
|
||||
|
|
|
@ -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:
|
||||
|
|
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue