fix typos and add warning messages when using --cache (#733)

Co-authored-by: Songtao Liu <ruinmessi@gmail.com>
Megvii-BaseDetection · Sep 27, 2021 · abc8c87 · abc8c87
1 parent 9a957f5
commit abc8c87
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/yolox/data/datasets/coco.py b/yolox/data/datasets/coco.py
@@ -100,7 +100,9 @@ def _cache_images(self):
             pbar.close()
         else:
             logger.warning(
-                "You are using cached imgs! Make sure your dataset is not changed!!"
+                "You are using cached imgs! Make sure your dataset is not changed!!\n"
+                "Everytime the self.input_size is changed in your exp file, you need to delete\n"
+                "the cached data and re-generate them.\n"
             )
 
         logger.info("Loading cached imgs...")

diff --git a/yolox/data/datasets/voc.py b/yolox/data/datasets/voc.py
@@ -152,7 +152,7 @@ def _cache_images(self):
         cache_file = self.root + "/img_resized_cache_" + self.name + ".array"
         if not os.path.exists(cache_file):
             logger.info(
-                "Caching images for the frist time. This might take about 3 minutes for VOC"
+                "Caching images for the first time. This might take about 3 minutes for VOC"
             )
             self.imgs = np.memmap(
                 cache_file,
@@ -175,7 +175,9 @@ def _cache_images(self):
             pbar.close()
         else:
             logger.warning(
-                "You are using cached imgs! Make sure your dataset is not changed!!"
+                "You are using cached imgs! Make sure your dataset is not changed!!\n"
+                "Everytime the self.input_size is changed in your exp file, you need to delete\n"
+                "the cached data and re-generate them.\n"
             )
 
         logger.info("Loading cached imgs...")