ssd_kerasV2 icon indicating copy to clipboard operation
ssd_kerasV2 copied to clipboard

Wrong Layer Choosen for SSD from MobilenetV2 backbone

Open prateethvnayak opened this issue 5 years ago • 2 comments

The line here corresponds to the 12th block output.

The correct replacement for it is FeatureExtractor = Model(inputs=mobilenetv2.input, outputs=mobilenetv2.get_layer('block_12_add').output)

prateethvnayak avatar Aug 20 '19 20:08 prateethvnayak

@prateethvnayak Thanks, your modification works. By the way, didn't you face this problem when you train your model?

InvalidArgumentError: 2 root error(s) found.
  (0) Invalid argument: Incompatible shapes: [16,7308,4] vs. [16,1692,4]
	 [[{{node loss/predictions_loss/sub}}]]
	 [[loss/mul/_3917]]
  (1) Invalid argument: Incompatible shapes: [16,7308,4] vs. [16,1692,4]
	 [[{{node loss/predictions_loss/sub}}]]
0 successful operations.
0 derived errors ignored.

wyjeong avatar Nov 07 '19 03:11 wyjeong

@wyjeong No, did you make some changes ? it seems to be an error in the loss module. I cannot comment more without looking at your code

prateethvnayak avatar Nov 14 '19 16:11 prateethvnayak