Spaces:
Running
Running
File size: 1,146 Bytes
de79343 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
import os.path as osp
import os
class parser(object):
def __init__(self):
self.name = "training_cloth_segm_u2net_exp1" # Expriment name
self.image_folder = "../imaterialist/train/" # image folder path
self.df_path = "../imaterialist/train.csv" # label csv path
self.distributed = False # True for multi gpu training
self.isTrain = True
self.fine_width = 192 * 4
self.fine_height = 192 * 4
# Mean std params
self.mean = 0.5
self.std = 0.5
self.batchSize = 2 # 12
self.nThreads = 2 # 3
self.max_dataset_size = float("inf")
self.serial_batches = False
self.continue_train = True
if self.continue_train:
self.unet_checkpoint = "prev_checkpoints/cloth_segm_unet_surgery.pth"
self.save_freq = 1000
self.print_freq = 10
self.image_log_freq = 100
self.iter = 100000
self.lr = 0.0002
self.clip_grad = 5
self.logs_dir = osp.join("logs", self.name)
self.save_dir = osp.join("results", self.name)
|