3bfbab7afd5850b4f21b73dd3184374f47eb1d98,models.py,,create_modules,#Any#Any#Any#,10

Before Change

            size = int(mdef["size"])
            stride = int(mdef["stride"]) if "stride" in mdef else (int(mdef["stride_y"]), int(mdef["stride_x"]))
            pad = (size - 1) // 2 if int(mdef["pad"]) else 0
            modules.add_module("Conv2d", nn.Conv2d(in_channels=output_filters[-1],
                                                   out_channels=filters,
                                                   kernel_size=size,
                                                   stride=stride,
                                                   padding=pad,
                                                   bias=not bn))
            if bn:
                modules.add_module("BatchNorm2d", nn.BatchNorm2d(filters, momentum=0.1))
            if mdef["activation"] == "leaky":  // TODO: activation study https://github.com/ultralytics/yolov3/issues/441
                modules.add_module("activation", nn.LeakyReLU(0.1, inplace=True))

After Change

Instances