Trial-faster-rcnn
阿新 • • 發佈：2021-12-11
目錄
motivation
試一下faster rcnn的程式碼, 主要想看看backbone訓練一部分好呢還是全部都訓練好呢?
實驗設定
| Attribute | Value |
|---|---|
| backbone | resnet50 |
| batch_size | 2 |
| batch_size_per_image | 256 |
| benchmark | True |
| beta1 | 0.9 |
| beta2 | 0.999 |
| dataset | voc2012 |
| description | FasterRCNNTrain=300-500-1024-5=default-sgd-0.005-0.0005=2=default |
| epochs | 15 |
| eval_freq | 5 |
| eval_train | False |
| eval_valid | True |
| featmap_names | 1,2,3,4 |
| learning_policy | default |
| log2console | True |
| log2file | True |
| log_path | ./logs/FasterRCNNTrain/voc2012-resnet50/FasterRCNNTrain=300-500-1024-5=default-sgd-0.005-0.0005=2=default-121022 |
| lr | 0.005 |
| max_size | 500 |
| min_size | 300 |
| momentum | 0.9 |
| optimizer | sgd |
| pretrained_name | paras.pt |
| pretrained_path | E:\dfk\pre_trained_backbones\resnet50 |
| progress | False |
| representation_size | 1024 |
| resume | False |
| seed | 1 |
| trainable_stages | 0 |
| transform | default |
| weight_decay | 0.0005 |
實驗結果
| trainable_stages | map(0.5) | map(0.75) |
|---|---|---|
| 0 | 0.6925 | 0.2299 |
| 3 | 0.7385 | 0.3368 |
| 5 | 0.7277 | 0.3719 |
看來全部不訓練是不行的, 還是微調一下比較好.
就是有一個問題, 我現在的與訓練部分是backbone, 如果和官方程式碼一樣用faster rcnn的預訓練結果是否也是如此.
我最初懷疑可能全部都訓練可能不會太好的原因是, backbone是預訓練過的, 但是後面的RPN, ROI都是隨機初始化的, 會不會頭尾太不均衡了, 現在看起來好像影響不是很大.