Hi,
im really new to neural networks and I try to start with a simple task: traffix sign detection (no classification). I really hope you can help me with this because in the internet I just find some classification solutions.
I use the dataset GTSDB (http://benchmark.ini.rub.de/?section=gtsdb&subsection=dataset 25). I don’t know if I get something wrong, but isn’t it possible to just get a boundary box of a interesting object without the information what object it is?
Also do I need to first train with the single traffic signs or can I train with the full sized images of a traffic scene?
At the moment my network look like this, but I changed it many times:
class TraffixSignDetection(Module):
    def __init__(self, input_shape, n_classes):
        super(TraffixSignDetection, self).__init__()
        
        self.layer1 = Sequential(
            Conv2d(3, 6, kernel_size=5, stride=1, padding=2),
            BatchNorm2d(6),
            ReLU(),
            MaxPool2d(kernel_size=2, stride=2))
        self.layer2 = Sequential(
            Conv2d(6, 12, kernel_size=5, stride=1, padding=2),
            BatchNorm2d(12),
            ReLU(),
            MaxPool2d(kernel_size=2, stride=2))
        
        self.activation = ReLU()
        self.fc1 = Linear(400, 120)
        # self.fc2 = Linear(120, n_classes)
    
    def forward(self, x):
        out = self.layer1(x)
        out = self.layer2(out)
        out = out.reshape(out.size(0), -1)
        out = self.fc1(out)
        out = self.activation(out)
        
        self.features = out

Since I’m new to this problems, I don’t know what information you need from me. But if you need something just ask. Thank you very much!