Refactor image input handling in QuickOutput components

c3sr · Jul 24, 2024 · 8056002 · 8056002
1 parent b20822a
commit 8056002
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/src/components/Experiment/QuickOutput/Outputs/ImageEnhancement/ImageEnhancement.js b/src/components/Experiment/QuickOutput/Outputs/ImageEnhancement/ImageEnhancement.js
@@ -13,7 +13,7 @@ export default function ImageEnhancement(props) {
     <SingleColumnImageOutput
       modelId={props.trial.model.id}
       duration={props.trial.results.duration}
-      image={<ImageEnhancementImage input={props.trial.inputs[0]}
+      image={<ImageEnhancementImage input={props.trial.inputs[0].src}
                                     feature={props.trial.results.responses[0].features[0]}/>}
 
     />

diff --git a/src/components/Experiment/QuickOutput/Outputs/ObjectDetection/ObjectDetection.js b/src/components/Experiment/QuickOutput/Outputs/ObjectDetection/ObjectDetection.js
@@ -20,7 +20,7 @@ export default function ObjectDetection(props) {
     if (sections.length === 0) return <NoPredictions modelId={props.trial.model.id}/>
 
     return <div className={getElement("top-row")}>
-      <ObjectDetectionImage img={props.trial.inputs[0]}
+      <ObjectDetectionImage img={props.trial.inputs[0].src}
                             filteredSections={filteredSections}
                             sections={sections}
                             hover={hover}

diff --git a/...periment/QuickOutput/Outputs/SemanticSegmentation/hooks/useSemanticSegmentationControl.js b/...periment/QuickOutput/Outputs/SemanticSegmentation/hooks/useSemanticSegmentationControl.js
@@ -10,7 +10,7 @@ export default function useSemanticSegmentationControl(trial) {
         labelToShow = feature?.semantic_segment.labels[hover.property] ?? "";
     }
 
-    const image = trial.inputs[0];
+    const image = trial.inputs[0].src;
     const {height, int_mask, labels, width} = feature?.semantic_segment ?? {
         height: 0,
         int_mask: [],

diff --git a/src/components/ExperimentDetails/Summaries/ImageEnhancementSummary.js b/src/components/ExperimentDetails/Summaries/ImageEnhancementSummary.js
@@ -9,7 +9,7 @@ export default function ImageEnhancementSummary(props) {
   return <div className={getBlock()}>
     <div className={getElement("summary")}>
       <div className={getElement("summary__content")}>
-        <ImageEnhancementImage input={props.trial.inputs[0]} feature={props.trial.results.responses[0].features[0]}/>
+        <ImageEnhancementImage input={props.trial.inputs[0].src} feature={props.trial.results.responses[0].features[0]}/>
       </div>
     </div>
   </div>

diff --git a/src/components/ExperimentDetails/Summaries/ObjectDetectionSummary.js b/src/components/ExperimentDetails/Summaries/ObjectDetectionSummary.js
@@ -16,7 +16,7 @@ export default function ObjectDetectionSummary(props) {
     <p className={getElement("header")}>Output:</p>
 
     <div className={getElement("results")}>
-      <ObjectDetectionImage img={props.trial.inputs[0]}
+      <ObjectDetectionImage img={props.trial.inputs[0].src}
                             filteredSections={filteredSections}
                             hover={hover}
                             confidence={filter.confidence}