Expected object of scalar type Byte but got scalar type Bool for argument #2 'other' about a-pytorch-tutorial-to-object-detection HOT 4 OPEN

Hello @tejalal ,
The problem occurs because the comparison is waiting for scalars but when using ">" you receive a Boolean tensor. I had the same problem with the latest release of Pytorch. I solved it by creating a variable that saves the comparison as Booleans and creating another tensor where the type of data is changed to uint8.

      condition = overlap[box] > max_overlap
      condition = torch.tensor(condition, dtype=torch.uint8).to(device)
      #print(torch.tensor(condition,dtype=torch.uint8))
      suppress = torch.max(suppress, condition)  

from a-pytorch-tutorial-to-object-detection.

Hello @tejalal ,
The problem occurs because the comparison is waiting for scalars but when using ">" you receive a Boolean tensor. I had the same problem with the latest release of Pytorch. I solved it by creating a variable that saves the comparison as Booleans and creating another tensor where the type of data is changed to uint8.

      condition = overlap[box] > max_overlap
      condition = torch.tensor(condition, dtype=torch.uint8).to(device)
      #print(torch.tensor(condition,dtype=torch.uint8))
      suppress = torch.max(suppress, condition)  

Hey @MartinEliasQ,
It gives UserWarning that uint8 is now deprecated, so I initialised suppress to torch.bool dtype and change following operation on suppress to logical operation. It gives no warning now.

                suppress = torch.zeros((n_above_min_score)).bool().to(device)  # (n_qualified)

                # Consider each box in order of decreasing scores
                for box in range(class_decoded_locs.size(0)):
                    # If this box is already marked for suppression
                    if suppress[box] == 1:
                        continue

                    # Suppress boxes whose overlaps (with this box) are greater than maximum overlap
                    # Find such boxes and update suppress indices
                    suppress = suppress | (overlap[box] > max_overlap)
                    # The max operation retains previously suppressed boxes, like an 'OR' operation

                    # Don't suppress this box, even though it has an overlap of 1 with itself
                    suppress[box] = 0

                # Store only unsuppressed boxes for this class
                image_boxes.append(class_decoded_locs[~suppress])
                image_labels.append(
                    torch.LongTensor(
                        (~suppress).sum().item() * [c]).to(device)
                )
                image_scores.append(class_scores[~suppress])

from a-pytorch-tutorial-to-object-detection.

Hello @tejalal ,
The problem occurs because the comparison is waiting for scalars but when using ">" you receive a Boolean tensor. I had the same problem with the latest release of Pytorch. I solved it by creating a variable that saves the comparison as Booleans and creating another tensor where the type of data is changed to uint8.

      condition = overlap[box] > max_overlap
      condition = torch.tensor(condition, dtype=torch.uint8).to(device)
      #print(torch.tensor(condition,dtype=torch.uint8))
      suppress = torch.max(suppress, condition)  

Hey @MartinEliasQ I'm still getting the same error even after trying your solution. Any tips?

from a-pytorch-tutorial-to-object-detection.

@jackson211 Do you know how to solve this problem?
Traceback (most recent call last):
File "eval.py", line 141, in
evaluate(test_loader, model, test_img_num)
File "eval.py", line 87, in evaluate
top_k=200, original_image = images, max_OCR_overlap=0.2, max_OCR_ratio = 0.2)
File "/ctpn/sroie/sroie/task1/model.py", line 588, in detect_objects
[original_image.width, original_image.height, original_image.width,
AttributeError: 'Tensor' object has no attribute 'width'

from a-pytorch-tutorial-to-object-detection.