YAML Metadata
Warning:
empty or missing yaml metadata in repo card
(https://huggingface.co/docs/hub/model-cards#model-card-metadata)
Training Tutorial
Prepare dataset
- put you images at
data_preprocess/tower_data_low_light
, only support png image now - cd folder
data_preprocess
- run command
python3 random_crop.py
Training Stage 1
- copy images from
data_preprocess/tower_data_low_light_crop
tocodes_SelfDACE/old_version/stage1/data/train_data/
- cd folder
codes_SelfDACE/old_version/stage1
- run command
python train_1stage.py
Training Stage 2
- copy checkpoint from
codes_SelfDACE/old_version/stage1/snapshots_light/your_checkpoint_from_stage_1
tocodes_SelfDACE/old_version/stage2/snapshots_light/pre_train_1.pth
- copy images from
data_preprocess/tower_data_low_light_crop
tocodes_SelfDACE/old_version/stage2/data/train_data/
- cd folder
codes_SelfDACE/old_version/stage2
- run command
python train_2stage.py