fix nonfinite
This commit is contained in:
parent
d2bace5a0e
commit
b0feca82d5
|
@ -23,10 +23,10 @@ Architecture:
|
|||
name: MobileNetV3
|
||||
scale: 0.5
|
||||
model_name: large
|
||||
disable_se: True
|
||||
disable_se: False
|
||||
Neck:
|
||||
name: DBFPN
|
||||
out_channels: 96
|
||||
out_channels: 256
|
||||
Head:
|
||||
name: DBHead
|
||||
k: 50
|
||||
|
@ -74,7 +74,7 @@ Train:
|
|||
channel_first: False
|
||||
- DetLabelEncode: # Class handling label
|
||||
- Resize:
|
||||
# size: [640, 640]
|
||||
size: [640, 640]
|
||||
- MakeBorderMap:
|
||||
shrink_ratio: 0.4
|
||||
thresh_min: 0.3
|
||||
|
|
|
@ -35,7 +35,7 @@ export1:null
|
|||
export2:null
|
||||
##
|
||||
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
|
||||
inference:tools/infer/predict_det.py
|
||||
--use_gpu:True|False
|
||||
|
|
Loading…
Reference in New Issue