Comments (4)
@shayamram If you want to use the same preprocessing that was used for validation- then yes padded rescale to 640X640 with BGR.
The dog was a square image anyway and the model handled it.
You can see the exact preprocessing that is used for validation for any recipe if you check their dataset_params, for instance these are the dataset_params that were used for training YoloX, and
notice the transforms part inside valid_dataset_params.
from super-gradients.
Join the discussion on DagsHub!
from super-gradients.
Hey @shayamram (:
I assume you are pointing to the "Predict" cell, notice the line:
image_tensor = torch.tensor(np.array(image)[:, :, ::-1].copy()).permute(2, 0, 1).unsqueeze(dim=0).float().cuda()
The [:, :, ::-1] part reverses the order of the 3rd dimension, and therefore converts it to BGR format.
I will add a comment to clarify that this is also being done there.
If you have any other questions, please let me know.
from super-gradients.
@shaydeci Thanks for you reply - so just to clarify: the model predict expects BGR and padded image? (example: the image was padded, I assume to maintain aspect ratio).
This seems different from the example here (the one with the dog). This example is RGB and only simple resize (no padding), so I just want to be extra-certain that the correct preprocessing for prediction is (resize+padding) in BGR, and not simple-resize (no padding) RGB as in the second example.
Thanks again
from super-gradients.
Related Issues (20)
- Correct image transforms for Yolo-NAS
- Work with keypoints for recognize some poses HOT 1
- Custom metrics that depends on image_path?
- DetectionRandomAffine target-size is in wrong format HOT 2
- COCO Recipe reporting low precision
- ImportError: cannot import name 'utils' from partially initialized module 'super_gradients.training' (most likely due to a circular import HOT 4
- yolo-nas-sat model availability
- AttributeError: 'RegSeg48' object has no attribute 'set_dataset_processing_params' HOT 1
- How to set different weight decay values for different modules of the model
- yolo nas pose demo/colab is broken
- How to get edge_links, edge_colors, keypoint_colors when using yolo nas pose onnx?
- Validation metrics = 0.0 during training yolo-nas
- YOLO NAS'S Precision is significantly lower compare to other later YOLO model even when using same dataset ? HOT 4
- BaseSGLogger storage_location parameter is systematically overriden, why?
- Access Joints Coordinate
- Ground tensor shape issue when training YOLO_NAS_S model on a custom dataset HOT 1
- Issue when training and predicting with a custom dataset and the YOLO_NAS_S model HOT 1
- Model training process halted for small dataset HOT 4
- Inquiry About Official Release Date of OBB Detection Models for YOLO-NAS and Training HOT 1
- ONNX Export Output has Incorrect Class Labels but Correct Box and Confidence HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from super-gradients.