Commit 8110b2e
fix: restore training data augmentation behavior for det model
1. Restore set_epoch_as_seed in SimpleDataSet to re-enable adaptive
shrink_ratio (0.4→0.6) curriculum learning in MakeBorderMap and
MakeShrinkMap across training epochs.
2. Remove default p=1.0 for Affine augmentation, restoring
albumentations default p=0.5 (50% rotation probability).
These two changes caused V5 dataset precision to drop significantly
(e.g. blur 0.904→0.709, printing_ch 0.926→0.708) while V4 remained
unaffected.
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>1 parent cbf6837 commit 8110b2e
2 files changed
Lines changed: 13 additions & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
158 | 158 | | |
159 | 159 | | |
160 | 160 | | |
161 | | - | |
162 | | - | |
163 | 161 | | |
164 | 162 | | |
165 | 163 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
51 | 51 | | |
52 | 52 | | |
53 | 53 | | |
| 54 | + | |
54 | 55 | | |
55 | 56 | | |
56 | 57 | | |
| |||
73 | 74 | | |
74 | 75 | | |
75 | 76 | | |
| 77 | + | |
| 78 | + | |
| 79 | + | |
| 80 | + | |
| 81 | + | |
| 82 | + | |
| 83 | + | |
| 84 | + | |
| 85 | + | |
| 86 | + | |
| 87 | + | |
| 88 | + | |
76 | 89 | | |
77 | 90 | | |
78 | 91 | | |
| |||
0 commit comments