| 123456789101112131415161718 |
- # 图块高度
- 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=32
- # pretrained
- PRETRAINED=True
- # 是否使用bias
- USE_BIAS=True
|