@@ -406,6 +406,7 @@ The extracted and upsampled COCO objects images and masks can be found in this [
406406
407407Please extract CelebAMask-HQ and 11k Hands images based on the splits found in [ drive] ( https://drive.google.com/drive/folders/15nZETWlGMdcKY6aHbchRsWkUI42KTNs5?usp=sharing ) .
408408
409+ mkdir data_materials
409410download file to ./data_materials
410411
411412``` none
@@ -426,13 +427,19 @@ apt-get install p7zip-full
426427
427428cd data_materials
428429
430+ # make occlusion-aware-face-dataset folder
431+ mkdir path-to-mmsegmentaion/data/occlusion-aware-face-dataset
432+
429433# extract celebAMask-HQ and split by train-set
430434unzip CelebAMask-HQ.zip
4314357za x CelebAMask-HQ-masks_corrected.7z -o./CelebAMask-HQ
432- # suggest better code if you have
436+ # copy training data to train-image-folder
433437rsync -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
435441xargs -n 1 -i echo {}.png < train.txt > mask_train.txt
442+ # copy training data to train-mask-folder
436443rsync -a ./CelebAMask-HQ/CelebAMask-HQ-masks_corrected/ --files-from=./mask_train.txt ./CelebAMask-HQ-WO-Train_mask
437444mv train.txt ../data/occlusion-aware-face-dataset
438445
@@ -454,7 +461,7 @@ mv coco_object/* .
454461
455462```
456463
457- ** Dataset Organization:**
464+ ** Dataset material Organization:**
458465
459466``` none
460467
@@ -478,31 +485,6 @@ mv coco_object/* .
478485│ ├── object_mask_x4
479486│ │ ├── {mask}.png
480487
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
506488```
507489
508490## Data Generation
@@ -548,4 +530,38 @@ SOURCE_DATASET.MASK_DIR "path/to/mmsegmentation/data_materials/CelebAMask-HQ-WO-
548530OCCLUDER_DATASET.IMG_DIR " path/to/jw93/mmsegmentation/data_materials/DTD/images"
549531```
550532
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+
551563<!-- #endregion -->
564+
565+ ``` python
566+
567+ ```
0 commit comments