code: def __init__(self, in_channels_list, out_channels, extra_blocks=None):
      self.inner_blocks, self.layer_blocks = [], []
      for in_channels in in_channels_list:
          self.inner_blocks.append(
              nn.Conv2d(in_channels, out_channels, kernel_size=1)
          )
          self.layer_blocks.append(
              nn.Conv2d(
                  out_channels, out_channels, kernel_size=3, padding=1
              )
          )
      self.extra_blocks = (
          ExtraFPNBlock(256, 256)
          if extra_blocks is None
          else extra_blocks
      )
  
imports: import math
  from tinygrad.helpers import flatten, get_child
  import tinygrad.nn
  from extra.models.resnet import ResNet
  import numpy
label_desc: Use image node_id 1 to create a new FPN object  with example: obj = FPN(in_channels_list, out_channels, extra_blocks)