AlessandroMondin · December 20, 2022 19:46
diff --git a/neck.py b/neck.py
 self.neck = nn.ModuleList()
 self.neck += [
    CBL(in_channels=first_out*16, out_channels=first_out*8, kernel_size=1, stride=1, padding=0),
    C3(in_channels=first_out*16, out_channels=first_out*8, width_multiple=0.25, depth=2, backbone=False),
    CBL(in_channels=first_out*8, out_channels=first_out*4, kernel_size=1, stride=1, padding=0),
    C3(in_channels=first_out*8, out_channels=first_out*4, width_multiple=0.25, depth=2, backbone=False),
    CBL(in_channels=first_out*4, out_channels=first_out*4, kernel_size=3, stride=2, padding=1),
    C3(in_channels=first_out*8, out_channels=first_out*8, width_multiple=0.5, depth=2, backbone=False),
    CBL(in_channels=first_out*8, out_channels=first_out*8, kernel_size=3, stride=2, padding=1),
    C3(in_channels=first_out*16, out_channels=first_out*16, width_multiple=0.5, depth=2, backbone=False)
 ]
	self.neck = nn.ModuleList()
	self.neck += [
	CBL(in_channels=first_out16, out_channels=first_out8, kernel_size=1, stride=1, padding=0),
	C3(in_channels=first_out16, out_channels=first_out8, width_multiple=0.25, depth=2, backbone=False),
	CBL(in_channels=first_out8, out_channels=first_out4, kernel_size=1, stride=1, padding=0),
	C3(in_channels=first_out8, out_channels=first_out4, width_multiple=0.25, depth=2, backbone=False),
	CBL(in_channels=first_out4, out_channels=first_out4, kernel_size=3, stride=2, padding=1),
	C3(in_channels=first_out8, out_channels=first_out8, width_multiple=0.5, depth=2, backbone=False),
	CBL(in_channels=first_out8, out_channels=first_out8, kernel_size=3, stride=2, padding=1),
	C3(in_channels=first_out16, out_channels=first_out16, width_multiple=0.5, depth=2, backbone=False)
	]