VITON-HD: Test Pre-trained Network and Sample Images

Test Pre-trained VITON-HD Network and Sample Images

This repository provides pre-trained networks and sample images from the test dataset. Please download *.pkl and test images from the VITON-HD Google Drive folder and unzip *.zip files. The test.py script assumes that the downloaded files are placed in the ./checkpoints/ and ./datasets/ directories.

Running the Test Script

python3 test.py --gpu 0 --pretrained_model ./checkpoints/viton-hd.pkl --in_path ./datasets/test/ --out_path ./outputs/

Viewing the Results

The results of the test can be seen in the output directory. The results include images containing the predicted visual attributes for each image. Additionally, a report.txt file containing the precision, recall, and F1 scores for each attribute is included.