diff --git a/src/otx/algorithms/common/adapters/mmcv/pipelines/load_image_from_otx_dataset.py b/src/otx/algorithms/common/adapters/mmcv/pipelines/load_image_from_otx_dataset.py index fa0c94518d9..564b69fb6e7 100644 --- a/src/otx/algorithms/common/adapters/mmcv/pipelines/load_image_from_otx_dataset.py +++ b/src/otx/algorithms/common/adapters/mmcv/pipelines/load_image_from_otx_dataset.py @@ -42,7 +42,10 @@ def _get_unique_key(results: Dict[str, Any]) -> Tuple: if "cache_key" in results: return results["cache_key"] d_item = results["dataset_item"] - results["cache_key"] = d_item.media.path, d_item.roi.id + if d_item.media.path: # when video extracted frames come, media.path is given by None + results["cache_key"] = d_item.media.path, d_item.roi.id + else: + results["cache_key"] = d_item.roi.id, d_item.annotation_scene.annotations[0].id return results["cache_key"] def _get_memcache_handler(self):