YOLOtiny.py 文件源码

def  predict(self,x):
        h = F.leaky_relu(self.c1(x),slope=0.1)
        h = F.max_pooling_2d(h,ksize=2,stride=2,pad=0)
        h = F.leaky_relu(self.c3(h),slope=0.1)
        h = F.max_pooling_2d(h,ksize=2,stride=2,pad=0)
        h = F.leaky_relu(self.c5(h),slope=0.1)
        h = F.max_pooling_2d(h,ksize=2,stride=2,pad=0)
        h = F.leaky_relu(self.c7(h),slope=0.1)
        h = F.max_pooling_2d(h,ksize=2,stride=2,pad=0)
        h = F.leaky_relu(self.c9(h),slope=0.1)
        h = F.max_pooling_2d(h,ksize=2,stride=2,pad=0)
        h = F.leaky_relu(self.c11(h),slope=0.1)
        h = F.max_pooling_2d(h,ksize=2,stride=2,pad=0)
        h = F.leaky_relu(self.c13(h),slope=0.1)
        h = F.leaky_relu(self.c14(h),slope=0.1)
        h = F.leaky_relu(self.c15(h),slope=0.1)
        h = F.leaky_relu(self.l16(h),slope=0.1)
        h = F.leaky_relu(self.l17(h),slope=0.1)
        # skip dropout
        h = self.l19(h)

        return h