@@ -406,6 +406,7 @@ The extracted and upsampled COCO objects images and masks can be found in this [
406
406
407
407
Please extract CelebAMask-HQ and 11k Hands images based on the splits found in [ drive] ( https://drive.google.com/drive/folders/15nZETWlGMdcKY6aHbchRsWkUI42KTNs5?usp=sharing ) .
408
408
409
+ mkdir data_materials
409
410
download file to ./data_materials
410
411
411
412
``` none
@@ -426,13 +427,19 @@ apt-get install p7zip-full
426
427
427
428
cd data_materials
428
429
430
+ # make occlusion-aware-face-dataset folder
431
+ mkdir path-to-mmsegmentaion/data/occlusion-aware-face-dataset
432
+
429
433
# extract celebAMask-HQ and split by train-set
430
434
unzip CelebAMask-HQ.zip
431
435
7za x CelebAMask-HQ-masks_corrected.7z -o./CelebAMask-HQ
432
- # suggest better code if you have
436
+ # copy training data to train-image-folder
433
437
rsync -a ./CelebAMask-HQ/CelebA-HQ-img/ --files-from=./CelebAMask-HQ-WO-train.txt ./CelebAMask-HQ-WO-Train_img
434
- basename -s .jpg ./CelebAMask-HQ-train/* > train.txt
438
+ # create a file-name txt file for copying mask
439
+ basename -s .jpg ./CelebAMask-HQ-WO-Train_img/* > train.txt
440
+ # add .png to file-name txt file
435
441
xargs -n 1 -i echo {}.png < train.txt > mask_train.txt
442
+ # copy training data to train-mask-folder
436
443
rsync -a ./CelebAMask-HQ/CelebAMask-HQ-masks_corrected/ --files-from=./mask_train.txt ./CelebAMask-HQ-WO-Train_mask
437
444
mv train.txt ../data/occlusion-aware-face-dataset
438
445
@@ -454,7 +461,7 @@ mv coco_object/* .
454
461
455
462
```
456
463
457
- ** Dataset Organization:**
464
+ ** Dataset material Organization:**
458
465
459
466
``` none
460
467
@@ -478,31 +485,6 @@ mv coco_object/* .
478
485
│ ├── object_mask_x4
479
486
│ │ ├── {mask}.png
480
487
481
- ├── data
482
- │ ├── occlusion-aware-face-dataset
483
- │ │ ├── train.txt
484
- │ │ ├── NatOcc_hand_sot
485
- │ │ │ ├── img
486
- │ │ │ │ ├── {image}.jpg
487
- │ │ │ ├── mask
488
- │ │ │ │ ├── {mask}.png
489
- │ │ ├── NatOcc_object
490
- │ │ │ ├── img
491
- │ │ │ │ ├── {image}.jpg
492
- │ │ │ ├── mask
493
- │ │ │ │ ├── {mask}.png
494
- │ │ ├── RandOcc
495
- │ │ │ ├── img
496
- │ │ │ │ ├── {image}.jpg
497
- │ │ │ ├── mask
498
- │ │ │ │ ├── {mask}.png
499
- │ │ ├── RealOcc
500
- │ │ │ ├── img
501
- │ │ │ │ ├── {image}.jpg
502
- │ │ │ ├── mask
503
- │ │ │ │ ├── {mask}.png
504
- │ │ │ ├── split
505
- │ │ │ │ ├── val.txt
506
488
```
507
489
508
490
## Data Generation
@@ -548,4 +530,38 @@ SOURCE_DATASET.MASK_DIR "path/to/mmsegmentation/data_materials/CelebAMask-HQ-WO-
548
530
OCCLUDER_DATASET.IMG_DIR " path/to/jw93/mmsegmentation/data_materials/DTD/images"
549
531
```
550
532
533
+ ** Dataset Organization:**
534
+
535
+ ``` none
536
+ ├── data
537
+ │ ├── occlusion-aware-face-dataset
538
+ │ │ ├── train.txt
539
+ │ │ ├── NatOcc_hand_sot
540
+ │ │ │ ├── img
541
+ │ │ │ │ ├── {image}.jpg
542
+ │ │ │ ├── mask
543
+ │ │ │ │ ├── {mask}.png
544
+ │ │ ├── NatOcc_object
545
+ │ │ │ ├── img
546
+ │ │ │ │ ├── {image}.jpg
547
+ │ │ │ ├── mask
548
+ │ │ │ │ ├── {mask}.png
549
+ │ │ ├── RandOcc
550
+ │ │ │ ├── img
551
+ │ │ │ │ ├── {image}.jpg
552
+ │ │ │ ├── mask
553
+ │ │ │ │ ├── {mask}.png
554
+ │ │ ├── RealOcc
555
+ │ │ │ ├── img
556
+ │ │ │ │ ├── {image}.jpg
557
+ │ │ │ ├── mask
558
+ │ │ │ │ ├── {mask}.png
559
+ │ │ │ ├── split
560
+ │ │ │ │ ├── val.txt
561
+ ```
562
+
551
563
<!-- #endregion -->
564
+
565
+ ``` python
566
+
567
+ ```
0 commit comments