app.py

app.py

@@ -16,6 +16,12 @@ for idx, images in enumerate(os.listdir(coco_dir)):
   image = os.path.join(coco_dir, images)
   if os.path.isfile(image) and idx < 10:
     coco_image.append(image)
+    
+_, dataset_info = tfds.load(
+    "coco/2017", split=["train", "validation", "test"], with_info=True,
+)
+#test_dataset = tfds.load("coco/2017", split="test", data_dir="data")
+int2str = dataset_info.features["objects"]["label"].int2str
 
 class AnchorBox:
     """Generates anchor boxes.