donnyyou / torchcv

TorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision
https://pytorchcv.com
Apache License 2.0
2.25k stars 378 forks source link

small bug about configure: trans_seq.padding and display_iters #121

Closed qrsforever closed 4 years ago

qrsforever commented 4 years ago

configs/cls/cifar/base_cifar10_cls.conf

         "trans_seq": ["random_hflip", "padding", "random_crop"],
         "padding": {
          "ratio": 1.0,
          "pad": [4, 4, 4, 4],
          "allow_outside_center": false
        },

       "display_iters": 20,

the code parse :

  1. pil_aug_transforms.py:

    PIL_AUGMENTATIONS_DICT = {
    'random_saturation': RandomSaturation,
    'random_hue': RandomHue,
    'random_perm': RandomPerm,
    'random_contrast': RandomContrast,
    'random_brightness': RandomBrightness,
    'random_gauss_blur': RandomGaussBlur,
    'random_hsv': RandomHSV,
    'random_pad': RandomPad,                      
    'random_border': RandomBorder,
    'random_hflip': RandomHFlip,               
    'random_resize': RandomResize,             
    'random_crop': RandomCrop,                 
    'random_focus_crop': RandomFocusCrop,      
    'random_det_crop': RandomDetCrop,          
    'random_resized_crop': RandomResizedCrop,  
    'random_rotate': RandomRotate,             
    'resize': Resize
    }

    padding to random_border

  2. image_classifier.py: 131

    # Print the log info & reset the states.                              
    if self.runner_state['iters'] % self.solver_dict['display_iter'] == 0:

    display_iters to display_iter

donnyyou commented 4 years ago

Thanks! You could submit a PR if convenient!