Skip to content

Commit bcc1604

Browse files
committed
added constrained network segmentation examples
1 parent 3538d91 commit bcc1604

File tree

5 files changed

+1344
-22
lines changed

5 files changed

+1344
-22
lines changed

examples/GeneralizedMinkowski/ConstrainedNeuralNetworkSegmentation_motorbike.jl

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -394,7 +394,6 @@ opt = Flux.ADAM(1f-3)
394394
fval2 = train(HN,image,image,P,active_channels,flip_dims,maxiter,opt)
395395

396396
maxiter = 150
397-
opt = Flux.ADAM(1f-3)
398397
flip_dims = []
399398
fval3 = train(HN,image,image,P,active_channels,flip_dims,maxiter,opt)
400399

@@ -450,7 +449,7 @@ savefig("prediction_bike.png")
450449
#Plot data+prediction
451450
figure(figsize=(5,4))
452451
imshow(Array(pred_thres)[2:end-1,2:end-1])
453-
imshow(image[2:end-1,2:end-1,1:3],alpha=0.6,cmap="Greys")
452+
imshow(image[2:end-1,2:end-1,1:3],alpha=0.6)
454453
ax = gca()
455454
ax[:xaxis][:set_visible](false)
456455
ax[:yaxis][:set_visible](false)

0 commit comments

Comments
 (0)