카테고리 없음

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 .