Hi
Have you solved this issue?
Can’t uderstand how data should look like if I have several objects for one class

  
      pytorch.org
  
  
    

torchvision.models — PyTorch 1.7.0 documentation 12



  
  
    
    
  
  



boxes (FloatTensor[N, 4]): the ground-truth boxes in [x1, y1, x2,y2] format, with values of x between 0 and W and values of y between 0 and H

labels (Int64Tensor[N]): the class label for each ground-truth box
So the model should take N boxes of size 4, where N is the number of classes, I can’t understand how to pass several objects of one class. Please help