Decathlon / Decathlon/decavision

Image scaling issues

Open
#32 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
4
Forks
3
PR merge metrics
No merged PRs in 30d

Description

1- Currently, I believe the images are not scaled to a [0-1] range when we run the tfrecords generation followed by model training. This is OK for efficientnet models, because [preprocessing is done by the model itself](https://keras.io/api/applications/efficientnet/#efficientnetb0-function). That said, other models, like [inception](https://keras.io/api/applications/inceptionv3/) expect a scaled input.

2- I might we wrong, but I believe the `scale` variable, used for data augmentation and created [here](https://github.com/Decathlon/decavision/blob/master/decavision/model_training/tfrecords_image_classifier.py#L110), should be the opposite. It should be `1` for models of the efficientnet family (because the model expect a tensor of [0-255] floats), while it should be 255 for the other models.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.