Deci-AI / super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.
https://www.supergradients.com
Apache License 2.0
4.6k stars 510 forks source link

SSD MobileNet V2 recipe #2002

Open Budelon opened 6 months ago

Budelon commented 6 months ago

💡 Your Question

I've read the docs on the COCO SSD recipe. The recipe train the model with

But it validates without the HSV transform. val_dataset_params: data_dir: /data/coco # root path to coco data subdir: images/val2017 # sub directory path of data_dir containing the train data. json_file: instances_val2017.json # path to coco train json file, data_dir/annotations/train_json_file. input_dim: [320, 320] cache_annotations: True ignore_empty_annotations: True transforms:

Questions:

  1. Why does the validation not use the HSV transform?
  2. If i want to run the model trained on COCO, should i transform the BGR image to HSV before running?

Thanks.

Versions

No response

BloodAxe commented 6 months ago

There is no mistake.A DetectionHSV is a augmentation transformation, which takes image in BGR colorspace, transform it to HSV colorspace and apply random shifts to H,S and V components and then convert image back to BGR. This is data augmentation transformation aimed to improve model robustness to color changes. On validation you don't want to have any sort of randomness.