카테고리 없음
deeplab 에러모음
DanGEE
2022. 5. 15. 19:37
1. ValueError: Total size of new array must be unchanged for MobilenetV2/Conv/weights lh_shape: [(3, 3, 3, 16)], rh_shape: [(3, 3, 3, 32)]
와 같은 에러 발생시.
deeplabv3에서 mnv2 모델 불러올때 문제인데,
해결법 :
common.py의 depth_multiplier 플래그를 1.0에서 0.5로 변경하여 해결한다. |
가중치 재사용에 관해
deeplab 학습시에 (xception, resnet에서 동작하는데, mobilenet에서도 잘 되는지는 모르겠다.
mobilenet에서는 false, false만 가능, 아니면 class개수 변경 불가
If you want to re-use all the trained weights, set initialize_last_layer=True.
If you want to re-use only the network backbone, set initialize_last_layer=False and last_layers_contain_logits_only=False.
If you want to re-use all the trained weights except the logits (since the num_classes may be different), set initialize_last_layer=False and last_layers_contain_logits_only=True .