fix nonfinite

This commit is contained in:
LDOUBLEV 2021-09-09 08:28:58 +00:00
parent d2bace5a0e
commit b0feca82d5
2 changed files with 4 additions and 4 deletions

View File

@ -23,10 +23,10 @@ Architecture:
name: MobileNetV3 name: MobileNetV3
scale: 0.5 scale: 0.5
model_name: large model_name: large
disable_se: True disable_se: False
Neck: Neck:
name: DBFPN name: DBFPN
out_channels: 96 out_channels: 256
Head: Head:
name: DBHead name: DBHead
k: 50 k: 50
@ -74,7 +74,7 @@ Train:
channel_first: False channel_first: False
- DetLabelEncode: # Class handling label - DetLabelEncode: # Class handling label
- Resize: - Resize:
# size: [640, 640] size: [640, 640]
- MakeBorderMap: - MakeBorderMap:
shrink_ratio: 0.4 shrink_ratio: 0.4
thresh_min: 0.3 thresh_min: 0.3

View File

@ -35,7 +35,7 @@ export1:null
export2:null export2:null
## ##
train_model:./inference/ch_ppocr_mobile_v2.0_det_train/best_accuracy train_model:./inference/ch_ppocr_mobile_v2.0_det_train/best_accuracy
infer_export:tools/export_model.py -c configs/det/det_mv3_db.yml -o infer_export:tools/export_model.py -c tests/configs/ch_ppocr_v2.0/ch_det_mv3_db_v2.0.yml -o
infer_quant:False infer_quant:False
inference:tools/infer/predict_det.py inference:tools/infer/predict_det.py
--use_gpu:True|False --use_gpu:True|False