Merge remote-tracking branch 'origin/main'

# Conflicts:
#	Frontal-View VTON/test.sh
#	Frontal-View VTON/train.sh
#	train.sh

Frontal-View VTON/test.sh

@@ -1,13 +1,29 @@
- CUDA_VISIBLE_DEVICES=3 python test_global_local.py --gpu_id 0 \
- --ddim_steps 50 \
- --outdir results/try\
- --config configs/viton512.yaml \
- --dataroot /mnt/pfs-mc0p4k/cvg/team/didonglin/why/datasets/VITON-HD \
- --ckpt checkpoints/viton-hd.ckpt \
- --n_samples 2 \
- --seed 23 \
- --scale 1 \
- --H 1024 \
- --W 768
+# CUDA_VISIBLE_DEVICES=3 python test_global_local.py --gpu_id 0 \
+# --ddim_steps 50 \
+# --outdir results/dresscode_gp_all_weights/ep6 \
+# --config configs/viton512.yaml \
+# --dataroot /mnt/pfs-mc0p4k/cvg/team/didonglin/why/datasets/VITON-HD \
+# --ckpt /mnt/pfs-mc0p4k/cvg/team/didonglin/why/DCI-VTON-Virtual-Try-On/models/dci-vton-only-full-cloth-wo-zero/2024-03-21T19-03-56_viton512/checkpoints/epoch=000006.ckpt \
+# --n_samples 2 \
+# --seed 23 \
+# --scale 1 \
+# --H 1024 \
+# --W 768
+
+#!/bin/bash
+# 循环遍历变量 i 从 0 到 40
+for ((i=10; i<100; i+=1)); do
+    CUDA_VISIBLE_DEVICES=7 python test_global_local.py --gpu_id 0 \
+    --ddim_steps 50 \
+    --outdir results/dresscode_gp_all_weights/ep$i \
+    --config configs/viton512.yaml \
+    --dataroot /mnt/pfs-mc0p4k/cvg/team/didonglin/why/datasets/Dresscode_upper_body_copy \
+    --ckpt /mnt/pfs-mc0p4k/cvg/team/didonglin/why/DCI-VTON-Virtual-Try-On/models/dci-vton-gp-dresscode-all-weights/2024-04-27T20-53-05_viton512/checkpoints/epoch=0000$i.ckpt \
+    --n_samples 1 \
+    --seed 23 \
+    --scale 1 \
+    --H 512 \
+    --W 384 
+done
 
 # pose

Frontal-View VTON/train.sh

@@ -1 +1,4 @@
-CUDA_VISIBLE_DEVICES=4,5 python -u main.py --logdir models/oc --pretrained_model checkpoints/model.ckpt --base configs/viton512.yaml --scale_lr False
+CUDA_VISIBLE_DEVICES=4,5 python -u main.py --logdir models/dci-vton-gp-dresscode-all-weights --pretrained_model /mnt/pfs-mc0p4k/cvg/team/didonglin/why/DCI-VTON-Virtual-Try-On-ca/checkpoints/viton512.ckpt --base configs/viton512.yaml --scale_lr False
+
+
+# code

train.sh

@@ -1 +1,3 @@
-CUDA_VISIBLE_DEVICES=4,5 python -u main.py --logdir models/oc --pretrained_model checkpoints/model.ckpt --base configs/viton512.yaml --scale_lr False
+# CUDA_VISIBLE_DEVICES=4,5 python -u main.py --logdir models/dci-vton-mv-warp-paper --pretrained_model /mnt/pfs-mc0p4k/cvg/team/didonglin/why/DCI-VTON-Virtual-Try-On-ca/checkpoints/viton512.ckpt --base configs/viton512.yaml --scale_lr False
+
+CUDA_VISIBLE_DEVICES=4,5 python -u main_finetune_part.py --logdir models/oc --pretrained_model /mnt/pfs-mc0p4k/cvg/team/didonglin/why/DCI-VTON-Virtual-Try-On-ca/checkpoints/viton512.ckpt --base configs/viton512.yaml --scale_lr False