# 图块高度 PATCH_WIDTH=256 # 图块宽度 PATCH_HEIGHT=256 # 测试精度时置信度阈值 CONFIDENCE_THRESHOLD=0.6 # 图片输入大小 IMG_INPUT_SIZE=256 # 工作线程数 WORKERS=0 # CUDA设备 CUDA_VISIBLE_DEVICES=0 # batch size BATCH_SIZE=128 # pretrained PRETRAINED=True
