From: @linqingke Reviewed-by: @yingjy,@oacjiewen Signed-off-by: @yingjytags/v1.1.0
| @@ -169,7 +169,6 @@ bash scripts/run_eval_ascend.sh $TRAINED_CKPT | |||||
| * `--TRAIN_DATASET_SIZE`:训练数据集大小。 | * `--TRAIN_DATASET_SIZE`:训练数据集大小。 | ||||
| * `--TEST_DATASET_PATH`:测试数据集的路径。 | * `--TEST_DATASET_PATH`:测试数据集的路径。 | ||||
| * `--TEST_BATCH_SIZE`:测试批次大小。 | * `--TEST_BATCH_SIZE`:测试批次大小。 | ||||
| * `--TEST_DATASET_SIZE`:测试数据集大小。 | |||||
| * `--TRAIN_EPOCHS`:总训练轮次。 | * `--TRAIN_EPOCHS`:总训练轮次。 | ||||
| * `--CKPT_PATH`:模型检查点文件路径,可用于恢复训练和评估。 | * `--CKPT_PATH`:模型检查点文件路径,可用于恢复训练和评估。 | ||||
| * `--SAVE_PATH`:模型检查点文件保存路径。 | * `--SAVE_PATH`:模型检查点文件保存路径。 | ||||
| @@ -22,7 +22,6 @@ config = ed({ | |||||
| "img_height": 768, | "img_height": 768, | ||||
| "keep_ratio": False, | "keep_ratio": False, | ||||
| "flip_ratio": 0.5, | "flip_ratio": 0.5, | ||||
| "photo_ratio": 0.5, | |||||
| "expand_ratio": 1.0, | "expand_ratio": 1.0, | ||||
| # anchor | # anchor | ||||
| @@ -109,7 +108,6 @@ config = ed({ | |||||
| "test_max_per_img": 100, | "test_max_per_img": 100, | ||||
| "test_batch_size": 2, | "test_batch_size": 2, | ||||
| "rpn_head_loss_type": "CrossEntropyLoss", | |||||
| "rpn_head_use_sigmoid": True, | "rpn_head_use_sigmoid": True, | ||||
| "rpn_head_weight": 1.0, | "rpn_head_weight": 1.0, | ||||
| @@ -118,7 +116,6 @@ config = ed({ | |||||
| "base_step": 58633, | "base_step": 58633, | ||||
| "total_epoch": 13, | "total_epoch": 13, | ||||
| "warmup_step": 500, | "warmup_step": 500, | ||||
| "warmup_mode": "linear", | |||||
| "warmup_ratio": 1/3.0, | "warmup_ratio": 1/3.0, | ||||
| "sgd_step": [8, 11], | "sgd_step": [8, 11], | ||||
| "sgd_momentum": 0.9, | "sgd_momentum": 0.9, | ||||
| @@ -39,7 +39,6 @@ config = ed({ | |||||
| "TRAIN_ROOT_DIR": 'psenet/ic15/', | "TRAIN_ROOT_DIR": 'psenet/ic15/', | ||||
| "TRAIN_IS_TRANSFORM": True, | "TRAIN_IS_TRANSFORM": True, | ||||
| "TRAIN_LONG_SIZE": 640, | "TRAIN_LONG_SIZE": 640, | ||||
| "TRAIN_DATASET_SIZE": 1000, | |||||
| "TRAIN_MIN_SCALE": 0.4, | "TRAIN_MIN_SCALE": 0.4, | ||||
| "TRAIN_BATCH_SIZE": 4, | "TRAIN_BATCH_SIZE": 4, | ||||
| "TRAIN_REPEAT_NUM": 1800, | "TRAIN_REPEAT_NUM": 1800, | ||||
| @@ -48,7 +47,6 @@ config = ed({ | |||||
| # dataset for test | # dataset for test | ||||
| "TEST_ROOT_DIR": 'psenet/ic15/', | "TEST_ROOT_DIR": 'psenet/ic15/', | ||||
| "TEST_DATASET_SIZE": 500, | |||||
| "TEST_BUFFER_SIZE": 4, | "TEST_BUFFER_SIZE": 4, | ||||
| "TEST_DROP_REMAINDER": False, | "TEST_DROP_REMAINDER": False, | ||||
| @@ -14,7 +14,6 @@ | |||||
| # ============================================================================ | # ============================================================================ | ||||
| cfg_unet = { | cfg_unet = { | ||||
| 'name': 'Unet', | |||||
| 'lr': 0.0001, | 'lr': 0.0001, | ||||
| 'epochs': 400, | 'epochs': 400, | ||||
| 'distribute_epochs': 1600, | 'distribute_epochs': 1600, | ||||