Context

This dataset contains information detected to objects detected in Shopee - Price Match Guarantee Train dataset. It is useful to load this information instead of detecting objects (15 min) everytime you want to check your CV score.

Content

The files contained in this dataset are the following:

• objects_names.txt: list of the 80 possible objects. Useful to get the object name from a prediction, since they are identified by index.
• train_obj_0X.csv: X means the confidence threshold. Check this notebook that outputs this file for further information about columns, among other information.
• train_obj_0X_one-hot.csv: same information as the previous file but in one-hot format. Check this notebook that generates the data

Acknowledgements

This dataset was generated using YOLOv5 model from Ultralytics, which github repo is available here Also, it uses the dataset created by myself to use this model without internet

Inspiration

Hope that this dataset helps others implement solutions related to objects and Shopee competition faster