Skip to content

[New Sample ] PyTorch Classification Sample with rocCV preprocessing#143

Open
paveltc wants to merge 6 commits intoROCm:developfrom
paveltc:pytorch_sample
Open

[New Sample ] PyTorch Classification Sample with rocCV preprocessing#143
paveltc wants to merge 6 commits intoROCm:developfrom
paveltc:pytorch_sample

Conversation

@paveltc
Copy link
Copy Markdown
Contributor

@paveltc paveltc commented Apr 3, 2026

Motivation

This is a PyTorch classification sample that demonstrates how to use rocCV to preprocess an image for running inference on in PyTorch using the resnet50 model.

Technical Details

rocCV needs to be built with Python3.11 to run this sample.

Test Plan

Make sure rocCV is built and installed properly with all Python tests passing.
Run the classification sample and check that the image is being classified correctly.
Use an image that is easy to check the classification of.

Test Result

I have tested the sample with a known image and the classification results were correct.

Submission Checklist

@codecov-commenter
Copy link
Copy Markdown

codecov-commenter commented Apr 8, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.

Additional details and impacted files
@@           Coverage Diff            @@
##           develop     #143   +/-   ##
========================================
  Coverage    74.31%   74.31%           
========================================
  Files           79       79           
  Lines         3355     3355           
  Branches       738      738           
========================================
  Hits          2493     2493           
  Misses         380      380           
  Partials       482      482           
🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

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.

3 participants