# You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. from __future__ import print_function from __future__ import unicode_literals from utils.collect import SegConfig import numpy as np cfg = SegConfig() ########################## 基本配置 ########################################### # 均值,图像预处理减去的均值 cfg.MEAN = [0.5, 0.5, 0.5] # 标准差,图像预处理除以标准差· cfg.STD = [0.5, 0.5, 0.5] # 批处理大小 cfg.BATCH_SIZE = 1 # 验证时图像裁剪尺寸(宽,高) cfg.EVAL_CROP_SIZE = tuple() # 训练时图像裁剪尺寸(宽,高) cfg.TRAIN_CROP_SIZE = tuple() # 多进程训练总进程数 cfg.NUM_TRAINERS = 1 # 多进程训练进程ID