@@ -221,7 +221,7 @@ Please use following command options when train and test:
221
221
--rgb-std 0.229,0.224,0.225
222
222
```
223
223
224
- ### Cutout, Random Erasing, mixup, RICAP, CutMix
224
+ ### Cutout, Random Erasing, mixup, RICAP, ICAP, CutMix
225
225
226
226
```
227
227
# Cutout
@@ -236,6 +236,9 @@ $ ./train.py example_images --mixup
236
236
# RICAP
237
237
$ ./train.py example_images --ricap
238
238
239
+ # ICAP
240
+ $ ./train.py example_images --icap
241
+
239
242
# CutMix
240
243
$ ./train.py example_images --cutmix
241
244
@@ -298,8 +301,9 @@ usage: train.py [-h] [--model ARCH] [--from-scratch] [--epochs EPOCHS]
298
301
[--random-erasing-r1 RANDOM_ERASING_R1]
299
302
[--random-erasing-r2 RANDOM_ERASING_R2] [--mixup]
300
303
[--mixup-alpha MIXUP_ALPHA] [--ricap]
301
- [--ricap-beta RICAP_BETA] [--ricap-with-line] [--cutmix]
302
- [--cutmix-beta CUTMIX_BETA] [--cutmix-p CUTMIX_P]
304
+ [--ricap-beta RICAP_BETA] [--ricap-with-line] [--icap]
305
+ [--icap-beta ICAP_BETA] [--cutmix] [--cutmix-beta CUTMIX_BETA]
306
+ [--cutmix-prob CUTMIX_PROB]
303
307
DIR
304
308
305
309
train
@@ -418,10 +422,14 @@ optional arguments:
418
422
--ricap-beta RICAP_BETA
419
423
RICAP beta (default: 0.3)
420
424
--ricap-with-line RICAP with boundary line (default: False)
425
+ --icap apply ICAP (default: False)
426
+ --icap-beta ICAP_BETA
427
+ ICAP beta (default: 0.3)
421
428
--cutmix apply CutMix (default: False)
422
429
--cutmix-beta CUTMIX_BETA
423
430
CutMix beta (default: 1.0)
424
- --cutmix-p CUTMIX_P CutMix probability (default: 1.0)
431
+ --cutmix-prob CUTMIX_PROB
432
+ CutMix probability (default: 1.0)
425
433
```
426
434
427
435
0 commit comments