-
Notifications
You must be signed in to change notification settings - Fork 34
Description
Hello, we encountered the following problems when we replaced Resnet101 with Mobilenet_v2. How can I solve it?
Traceback (most recent call last):
File "train.py", line 185, in
train()
File "train.py", line 47, in train
gtboxes_batch=gtboxes_and_label)
File "../libs/networks/build_whole_network.py", line 356, in build_whole_detection_network
feature_to_cropped = self.build_base_network(input_img_batch)
File "../libs/networks/build_whole_network.py", line 38, in build_base_network
return mobilenet_v2.mobilenetv2_base(input_img_batch, is_training=self.is_training)
File "../libs/networks/mobilenet_v2.py", line 109, in mobilenetv2_base
finegrain_classification_mode=False)
File "/home/ubuntu/.local/lib/python3.6/site-packages/tensorflow/contrib/framework/python/ops/arg_scope.py", line 182, in func_with_args
return func(*args, **current_args)
File "../libs/networks/mobilenet/mobilenet_v2.py", line 162, in mobilenet_base
base_only=True, **kwargs)
File "/home/ubuntu/.local/lib/python3.6/site-packages/tensorflow/contrib/framework/python/ops/arg_scope.py", line 182, in func_with_args
return func(*args, **current_args)
File "../libs/networks/mobilenet/mobilenet_v2.py", line 154, in mobilenet
**kwargs)
File "/home/ubuntu/.local/lib/python3.6/site-packages/tensorflow/contrib/framework/python/ops/arg_scope.py", line 182, in func_with_args
return func(*args, **current_args)
File "../libs/networks/mobilenet/mobilenet.py", line 327, in mobilenet
net, end_points = mobilenet_base(inputs, scope=scope, **mobilenet_args)
File "/home/ubuntu/.local/lib/python3.6/site-packages/tensorflow/contrib/framework/python/ops/arg_scope.py", line 182, in func_with_args
return func(*args, **current_args)
File "../libs/networks/mobilenet/mobilenet.py", line 200, in mobilenet_base
_set_arg_scope_defaults(conv_defs_defaults),
File "/usr/lib/python3.6/contextlib.py", line 81, in enter
return next(self.gen)
File "../libs/networks/mobilenet/mobilenet.py", line 91, in _set_arg_scope_defaults
func, default_arg = items[0]
TypeError: 'dict_items' object does not support indexing