Update README.md
Browse files
README.md
CHANGED
|
@@ -28,7 +28,17 @@ The working dataset is a subset of this,
|
|
| 28 |
consisting of 9,616 images across 200 grocery product classes, sourced via
|
| 29 |
Roboflow
|
| 30 |
([universe.roboflow.com/groceries-jxjfd/grocery-goods](https://universe.roboflow.com/groceries-jxjfd/grocery-goods)).
|
| 31 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
([https://app.roboflow.com/bdata-497-advanced-topics-in-dv-nqagm/grocery-goods-ezyyb](https://universe.roboflow.com/bdata-497-advanced-topics-in-dv-nqagm/grocery-goods-ezyyb)).
|
| 33 |
|
| 34 |
### Class Distribution
|
|
@@ -53,6 +63,29 @@ This is the final dataset used for training, after the annotation process.
|
|
| 53 |
| gum | 1,923 | 1,346 | 385 | 192 |
|
| 54 |
| stationery | 1,466 | 1,026 | 293 | 147 |
|
| 55 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
## Training Procedure
|
| 57 |
|
| 58 |
- **Architecture**: YOLOv11n
|
|
|
|
| 28 |
consisting of 9,616 images across 200 grocery product classes, sourced via
|
| 29 |
Roboflow
|
| 30 |
([universe.roboflow.com/groceries-jxjfd/grocery-goods](https://universe.roboflow.com/groceries-jxjfd/grocery-goods)).
|
| 31 |
+
|
| 32 |
+
### Annotation Process
|
| 33 |
+
|
| 34 |
+
The original RPC-Dataset contained 200 product-specific classes,
|
| 35 |
+
where each class represented a specific product variant (e.g., `100_milk`, `101_milk`, `102_milk`). These classes were collapsed into 17 broader product categories
|
| 36 |
+
to improve generalization, reduce class imbalance, and better reflect how a self-checkout system categorizes items by type rather than specific SKU.
|
| 37 |
+
For example, all milk classes were merged into a single `milk` class, reducing the total class count from 200 to 17.
|
| 38 |
+
Random samples were reviewed after relabeling to validate annotation quality, with no corrections needed.
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
This is the final dataset used for training, after the annotation process
|
| 42 |
([https://app.roboflow.com/bdata-497-advanced-topics-in-dv-nqagm/grocery-goods-ezyyb](https://universe.roboflow.com/bdata-497-advanced-topics-in-dv-nqagm/grocery-goods-ezyyb)).
|
| 43 |
|
| 44 |
### Class Distribution
|
|
|
|
| 63 |
| gum | 1,923 | 1,346 | 385 | 192 |
|
| 64 |
| stationery | 1,466 | 1,026 | 293 | 147 |
|
| 65 |
|
| 66 |
+
### Train/Validation/Test Split
|
| 67 |
+
| Split | Ratio | Count |
|
| 68 |
+
| ---------- | ----- | ------ |
|
| 69 |
+
| Train | 70% | 36,928 |
|
| 70 |
+
| Validation | 20% | 10,505 |
|
| 71 |
+
| Test | 10% | 5,276 |
|
| 72 |
+
|
| 73 |
+
### Data Augmentation
|
| 74 |
+
The following augmentations were applied during training to simulate real-world checkout conditions:
|
| 75 |
+
| Augmentation | Purpose |
|
| 76 |
+
| ---------------------------------- | --------------------------------------- |
|
| 77 |
+
| Rotation | Items placed on belt in any orientation |
|
| 78 |
+
| Horizontal/Vertical Flip | Additional orientation variation |
|
| 79 |
+
| Mosaic | Multiple items on belt simultaneously |
|
| 80 |
+
| HSV Shift (hue, saturation, value) | Simulate varied store lighting |
|
| 81 |
+
| Translation & Scale | Camera height and position variation |
|
| 82 |
+
|
| 83 |
+
### Known Biases and Limitations
|
| 84 |
+
- Dataset is predominantly composed of Chinese grocery product packaging, limiting generalizability to Western or European retail environments
|
| 85 |
+
- Fresh and unpackaged produce (such as fruits or vegetables) are not represented in the dataset
|
| 86 |
+
- Limited lighting variation — real checkout environments may have inconsistent lighting not well represented in training images
|
| 87 |
+
|
| 88 |
+
|
| 89 |
## Training Procedure
|
| 90 |
|
| 91 |
- **Architecture**: YOLOv11n
|