full image segmentation checkpoint

Author: seveibar

src/Annotator/index.js

@@ -63,6 +63,7 @@ export const Annotator = ({
   videoTime = 0,
   videoName,
   onExit,
+  mask,
   onNextImage,
   onPrevImage,
 }: Props) => {
@@ -135,6 +136,8 @@ export const Annotator = ({
         RegionEditLabel={RegionEditLabel}
         alwaysShowNextButton={Boolean(onNextImage)}
         alwaysShowPrevButton={Boolean(onPrevImage)}
+        mask={mask}
+        // maskVersion={maskVersion}
         state={state}
         dispatch={dispatch}
       />

src/FullImageSegmentationAnnotator/index.js

@@ -2,6 +2,23 @@
 import React from "react"
 import Annotator from "../Annotator"
 
+const [width, height] = [200, 200]
+
+// TODO remove this !!!!
+const uint8Array = new Uint8ClampedArray(4 * width * height)
+
+for (let ri = 0; ri < width; ri++) {
+  for (let ci = 0; ci < height; ci++) {
+    uint8Array[(ri * width + ci) * 4 + 0] = Math.floor(Math.random() * 256)
+    uint8Array[(ri * width + ci) * 4 + 1] = Math.floor(Math.random() * 256)
+    uint8Array[(ri * width + ci) * 4 + 2] = Math.floor(Math.random() * 256)
+    uint8Array[(ri * width + ci) * 4 + 3] = 255
+  }
+}
+
+const imageData = new ImageData(uint8Array, width, height)
+// TODO remove this !!!!
+
 export default ({ onExit, images }) => {
-  return <Annotator images={images} onExit={onExit} />
+  return <Annotator mask={imageData} images={images} onExit={onExit} />
 }

src/ImageCanvas/index.js

@@ -66,7 +66,7 @@ type Props = {
 
 const getDefaultMat = () => Matrix.from(1, 0, 0, 1, -10, -10)
 
-export default ({
+export const ImageCanvas = ({
   regions,
   imageSrc,
   videoSrc,
@@ -480,3 +480,5 @@ export default ({
     </div>
   )
 }
+
+export default ImageCanvas

src/ImageMask/index.story.js

@@ -9,7 +9,7 @@ import ImageMask from "./"
 
 const [width, height] = [200, 200]
 
-const uint8Array = new Uint8ClampedArray(4 * width * width)
+const uint8Array = new Uint8ClampedArray(4 * width * height)
 
 for (let ri = 0; ri < width; ri++) {
   for (let ci = 0; ci < height; ci++) {

src/MainLayout/index.js

@@ -36,11 +36,12 @@ type Props = {
   alwaysShowPrevButton?: boolean,
 }
 
-export default ({
+export const MainLayout = ({
   state,
   dispatch,
   alwaysShowNextButton = false,
   alwaysShowPrevButton = false,
+  mask,
   RegionEditLabel,
 }: Props) => {
   const classes = useStyles()
@@ -154,6 +155,7 @@ export default ({
                 <ImageCanvas
                   {...settings}
                   key={state.selectedImage}
+                  mask={mask}
                   showTags={state.showTags}
                   allowedArea={state.allowedArea}
                   regionClsList={state.regionClsList}
@@ -262,3 +264,5 @@ export default ({
     </Fullscreen>
   )
 }
+
+export default MainLayout