Skip to content
This repository has been archived by the owner on Aug 19, 2023. It is now read-only.

Debug changes #210

Open
wants to merge 3 commits into
base: master
Choose a base branch
from
Open

Debug changes #210

wants to merge 3 commits into from

Conversation

ErezYosef
Copy link

@ErezYosef ErezYosef commented Feb 10, 2021

Changes:

  • cv2 read images in BGR format. The model trained in RGB input, so in inference it should be the same color space.

  • The indexing of scores should be the same as classification and transformed_anchors.
    Otherwise, the presented score will not match the right prediction box.

  • Removed unnecessary lines of code in the model. the finalResult list has no further use in the code. It might be there because of previous code versions.

@ErezYosef ErezYosef changed the title fixes to visualize_single_image fixes and changes Feb 10, 2021
@ErezYosef ErezYosef changed the title fixes and changes Debug changes Feb 10, 2021
@ErezYosef
Copy link
Author

ErezYosef commented Feb 10, 2021

  • CPU compliance fix: Change image.coda().float() to image.float()
    There is a few lines above the necessary code:
            if torch.cuda.is_available():
                image = image.cuda()

Fixes issue mentioned: #200 (comment)

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant