"Wrong or unsupported dataset. Available 'COCO' or 'VOC'")
print("\n\Testing on %s dataset" % (DATASET_NAME + TESTSET_YEAR))
dataset.show_info()
_ = input("Press Enter to continue...")
## 2. Dataloader initialization
print("\t2. Dataloader initialization...")
dataloader = Dataloader(dataset, TEST_SIZE)
test_generator = dataloader.generate_batch("test")
## 3. Network initialization
print("\t3. Network initialization...")
ssd = SSD(num_classes=len(dataset.label_ids) + 1, input_shape=INPUT_SHAPE)
latest = tf.train.latest_checkpoint(CHECKPOINT_DIR)
ssd.load_weights(latest)
ssd.summary()
_ = input("Press Enter to continue...")
## 4. Generate default boxes
print("\t4. Default boxes generation...")
fm_shapes = ssd.output_shape
aspect_ratios = ASPECT_RATIOS
scales = SCALES
default_boxes = Image.generate_default_boxes(fm_shapes, aspect_ratios,
scales)
# ---------------------------------------------------------------- #
print("Initialization completed!")
print("Start Testing...")
# -------------------------- Test loop -------------------------- #
