def test_yolov2_tiny_2(self): from dlpy.applications import Tiny_YoloV1 model = Tiny_YoloV1(self.s, n_classes=7) model.print_summary()
def test_yolov1_tiny(self): from dlpy.applications import Tiny_YoloV1 model = Tiny_YoloV1(self.s) model.print_summary()