search

LZHMS / Virtual-Tryon

基于深度学习的2D虚拟试衣并进行模型压缩优化
6 stars 1 forks source link

'--label_path'指的什么 save_checkpoint是什么 #2

Open itzok6744 opened 1 year ago

itzok6744 commented 1 year ago

1

base_options.py文件中， self.parser.add_argument('--save_path', type=str, default='D:/Pycharm/Virtual-Tryon-main/images/示例图片/group1/test/01.jpg', help='save the try_on image') self.parser.add_argument('--image_path', type=str, default='D:/Pycharm/Virtual-Tryon-main/images/示例图片/group1/images/000266_0.jpg', help='images path') self.parser.add_argument('--clothe_path', type=str, default='D:/Pycharm/Virtual-Tryon-main/images/示例图片/group1/clothes/000220_1.jpg', help='clothes path') self.parser.add_argument('--label_path', type=str, default='dataset/train_label', help='labels path') self.parser.add_argument('--save_checkpoint', type=str, default='./checkpoints/Pruning', help='save the prunned model')

LZHMS commented 1 year ago

--label_path表示目标图像的路径，save_checkpoint参数表示模型剪枝和压缩后的存放路径