fix nonfinite
This commit is contained in:
parent
d2bace5a0e
commit
b0feca82d5
|
@ -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
|
||||||
|
|
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue