Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Application] cuda support for example of pytorch yolo v2 @open sesame 05/10 19:54 #2575

Merged
merged 1 commit into from
May 20, 2024

Conversation

baek2sm
Copy link
Contributor

@baek2sm baek2sm commented May 8, 2024

  • add cuda option to train yolo v2 model backbone (for converting to nntrainer model)
  • preprocessing for input dataset
    • unmatched paired dataset
    • no annotation value

Signed-off-by: Seungbaek Hong [email protected]

@taos-ci
Copy link

taos-ci commented May 8, 2024

📝 TAOS-CI Version: 1.5.20200925. Thank you for submitting PR #2575. Please a submit 1commit/1PR (one commit per one PR) policy to get comments quickly from reviewers. Your PR must pass all verificiation processes of cibot before starting a review process from reviewers. If you are new member to join this project, please read manuals in documentation folder and wiki page. In order to monitor a progress status of your PR in more detail, visit http://ci.nnstreamer.ai/.

@taos-ci
Copy link

taos-ci commented May 8, 2024

Copy link

@taos-ci taos-ci left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@baek2sm, 💯 All CI checkers are successfully verified. Thanks.

@baek2sm baek2sm changed the title [Wait for #2571][Application] cuda support for example of pytorch yolo v2 [Wait for #2571][Application] cuda support for example of pytorch yolo v2 @open sesame 05/09 13:21 May 9, 2024
@taos-ci
Copy link

taos-ci commented May 9, 2024

@baek2sm baek2sm changed the title [Wait for #2571][Application] cuda support for example of pytorch yolo v2 @open sesame 05/09 13:21 [Wait for #2571][Application] cuda support for example of pytorch yolo v2 @open sesame 05/09 13:24 May 9, 2024
@taos-ci
Copy link

taos-ci commented May 9, 2024

Copy link

@taos-ci taos-ci left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@baek2sm, 💯 All CI checkers are successfully verified. Thanks.

@baek2sm baek2sm changed the title [Wait for #2571][Application] cuda support for example of pytorch yolo v2 @open sesame 05/09 13:24 [Wait for #2571][Application] cuda support for example of pytorch yolo v2 @open sesame 05/10 19:54 May 10, 2024
@taos-ci
Copy link

taos-ci commented May 10, 2024

@taos-ci
Copy link

taos-ci commented May 10, 2024

:octocat: cibot: @baek2sm, A builder checker could not be completed because one of the checkers is not completed. In order to find out a reason, please go to http://ci.nnstreamer.ai/nntrainer/ci/repo-workers/pr-checker/2575-202405101953420.2483069896698-847cffad2e54cd3dd6a0705d5630d4e6782a2bcc/.

Copy link
Member

@SeoHyungjun SeoHyungjun left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

self.input_images = []
self.bbox_gt = []
self.cls_gt = []

for i in range(len(img_list)):
img = np.array(Image.open(img_list[i]).resize((416, 416))) / 255
for i in range(len(ids_list)):
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Don't need to use i in this for loop?
I think can change the for loop to use the elements directly.

for ids in ids_lilst:
    ...
    with open(ann_dir + ids + ".txt", "rt", encoding="utf-8") as f:
        ...

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@SeoHyungjun right. thanks! i'll apply it

@taos-ci
Copy link

taos-ci commented May 17, 2024

- add cuda option to train yolo v2 model backbone
- preprocessing for input dataset
  * unmatched paired dataset
  * no annotation value

Signed-off-by: Seungbaek Hong <[email protected]>
@taos-ci
Copy link

taos-ci commented May 17, 2024

Copy link

@taos-ci taos-ci left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@baek2sm, 💯 All CI checkers are successfully verified. Thanks.

@baek2sm baek2sm changed the title [Wait for #2571][Application] cuda support for example of pytorch yolo v2 @open sesame 05/10 19:54 [Application] cuda support for example of pytorch yolo v2 @open sesame 05/10 19:54 May 17, 2024
@taos-ci
Copy link

taos-ci commented May 17, 2024

Copy link

@taos-ci taos-ci left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@baek2sm, 💯 All CI checkers are successfully verified. Thanks.

Copy link
Collaborator

@jijoongmoon jijoongmoon left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@jijoongmoon jijoongmoon merged commit 6f817c7 into nnstreamer:main May 20, 2024
33 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants