Class Light_Classifier

light_classifier.py:20–59

# import pandas as pd

class Light_Classifier(torch.nn.Module):
   def __init__(self):
      super(Light_Classifier, self).__init__()
      self.conv1 = self.conv_block(c_in = 3, c_out = 15, kernel_size = 3, stride = 1, padding = 1)
      self.conv2 = self.conv_block(c_in = 15, c_out = 12, kernel_size = 3, stride = 1, padding = 1)
      self.conv3 = self.conv_block(c_in = 12, c_out = 3, kernel_size = 3, stride = 1, padding = 1)
      # 32px --> 48
      # 100px --> 432
      # self.bigN = 432
      self.bigN = 48
      self.fc1 = nn.Linear(self.bigN, 32)
      self.fc2 = nn.Linear(32, 16)
      self.fc3 = nn.Linear(16, 1)
      self.maxpool = nn.MaxPool2d(kernel_size=2, stride=2)
      # self.prefc1 = nn.Linear(self.bigN, )

   def conv_block(self, c_in, c_out, dropout=0.1, kernel_size=3, stride=1, **kwargs):
      seq_block = nn.Sequential(
         nn.Conv2d(in_channels=c_in, out_channels=c_out, kernel_size=kernel_size, **kwargs),
         nn.BatchNorm2d(num_features=c_out),
         nn.ReLU(),
         # nn.Dropout2d(p=dropout)
      )
      return seq_block

   def forward(self, x):
      x = self.conv1(x)
      x = self.maxpool(x)

      x = self.conv2(x)
      x = self.maxpool(x)

      x = self.conv3(x)
      x = self.maxpool(x)
      x = x.reshape((-1, self.bigN))

      x = F.relu(self.fc1(x))
      x = F.relu(self.fc2(x))
      x = torch.sigmoid(self.fc3(x))
      return x

class Heavy_Classifier(torch.nn.Module):
   def __init__(self):