CNN Prediction output to masked image - computer-vision

Question

Given an output prediction of shape [1,21,388,88] from my Unet. How can I plot it as a masked image? I am using PASCAL dataset.
Thanks!

Answer 1

Basically you can do something like this and give conditional so when they meet requirements you get the style/ image you want
<template>
<menu>
<router-link to="/">
<li>
<div :class="toggleData ? 'style-one' : 'style-two'"> <!-- Check if toggledata true we use style-one css, if not style-two -->
<img :src="toggleData ? 'image-one-link' : 'image-link-two'" /> <!-- Check if toggledata true we use image-one-link, if not image-two-link -->
</div>
<div class="name" >Home</div>
</li>
</router-link>
<router-link to="/distination2">
<li>
<div class="Pic">
<img src='' />
</div>
<div class="Name">Home2</div>
</li>
</router-link>
<router-link to="/distination3">
<li>
<div class="Pic">
<img src='' />
</div>
<div class="Name">Home3</div>
</li>
</router-link>
<button #click="buttonClick"> Click Me </button> <!-- When button click we change toggleData value to true -->
</menu>
</template>
<script>
export default {
data() {
return {
toggleData: false,
};
},
methods: {
buttonClick() {
this.toggleData = true;
},
},
}
</script>
<style scoped>
.style-one {
background: red;
}
.style-two {
background: blue;
}
</style>