def test_yolov2_multi(self): from dlpy.applications import YoloV2_MultiSize anchors = [] anchors.append(1) anchors.append(1) anchors.append(1) anchors.append(1) model = YoloV2_MultiSize(self.s, anchors, predictions_per_grid=2) model.print_summary()
def test_yolov2_multi_2(self): from dlpy.applications import YoloV2_MultiSize anchors=[] anchors.append(1) anchors.append(1) anchors.append(1) anchors.append(1) model = YoloV2_MultiSize(self.s, anchors, predictions_per_grid=2, max_label_per_image=3, max_boxes=1) model.print_summary()