I have also applied your solution and got acc > 0 but still it is very low and i am using resnet34 for my detection task. And i have tried few samples to check if my data is correct and there is no problem in my data
and here is the link for my post.

  
    
    
    Error is increasing in Single Object detection vision
  
  
    Hi, I am trying to detect the largest item(Single object detection) and I have used resnet34. 
and I am using L1 loss for bounding box and Cross-Entropy for classification but my loss in training is quite low around 20-30 but invalidation the loss is above 200. And Accuracy is always 0 
Any suggestions on how can I solve this problem. 
`Custom head for resnet34 -



custom_fc = nn.Sequential(
Flatten(),
nn.ReLU(),
nn.Dropout(0.5),
nn.Linear(512*7*7, 256),
nn.ReLU(),
nn.BatchNorm1d(256),
nn.Dropo…
  


Any suggestions why its happening