Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Small ConvolutionalIterationListener improvements (#382) 

* Add a message to the runtime exception

Signed-off-by: Paul Dubs <paul.dubs@gmail.com>

* Output Convolutions as PNG instead of JPG

A lossless encoding is useful in this case, as it allows small details to be preserved

Signed-off-by: Paul Dubs <paul.dubs@gmail.com>
master
Paul Dubs 2020-04-17 02:19:21 +02:00 committed by GitHub
parent 3d15706ffa
commit bb9cdb251e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deeplearning4j/deeplearning4j-ui-parent/deeplearning4j-ui/src/main/java/org/deeplearning4j/ui/weights/ConvolutionalIterationListener.java
View File

@ -140,7 +140,7 @@ public class ConvolutionalIterationListener extends BaseTrainingListener {
ComputationGraph l = (ComputationGraph) model; ComputationGraph l = (ComputationGraph) model;
Layer[] layers = l.getLayers(); Layer[] layers = l.getLayers();
if(layers.length != activations.size()) if(layers.length != activations.size())
throw new RuntimeException(); throw new RuntimeException("layers.length != activations.size(). Got layers.length="+layers.length+", activations.size()="+activations.size());
for( int i=0; i<layers.length; i++ ){ for( int i=0; i<layers.length; i++ ){
if(layers[i].type() == Layer.Type.CONVOLUTIONAL){ if(layers[i].type() == Layer.Type.CONVOLUTIONAL){
String layerName = layers[i].conf().getLayer().getLayerName(); String layerName = layers[i].conf().getLayer().getLayerName();

8
deeplearning4j/deeplearning4j-ui-parent/deeplearning4j-vertx/src/main/java/org/deeplearning4j/ui/module/convolutional/ConvolutionalListenerModule.java
View File

@ -105,22 +105,22 @@ public class ConvolutionalListenerModule implements UIModule {
BufferedImage bi = clp.getImg(); BufferedImage bi = clp.getImg();
ByteArrayOutputStream baos = new ByteArrayOutputStream(); ByteArrayOutputStream baos = new ByteArrayOutputStream();
try { try {
ImageIO.write(bi, "jpg", baos); ImageIO.write(bi, "png", baos);
} catch (IOException e) { } catch (IOException e) {
log.warn("Error displaying image", e); log.warn("Error displaying image", e);
} }
rc.response() rc.response()
.putHeader("content-type", "image/jpg") .putHeader("content-type", "image/png")
.end(Buffer.buffer(baos.toByteArray())); .end(Buffer.buffer(baos.toByteArray()));
} else { } else {
rc.response() rc.response()
.putHeader("content-type", "image/jpg") .putHeader("content-type", "image/png")
.end(Buffer.buffer(new byte[0])); .end(Buffer.buffer(new byte[0]));
} }
} else { } else {
rc.response() rc.response()
.putHeader("content-type", "image/jpg") .putHeader("content-type", "image/png")
.end(Buffer.buffer(new byte[0])); .end(Buffer.buffer(new byte[0]));
} }
} }