OlafenwaMoses / ImageAI

A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities
https://www.genxr.co/#products
MIT License
8.56k stars 2.19k forks source link

Model load error for RetinaNet #210

Open Aban8 opened 5 years ago

Aban8 commented 5 years ago

Hi,

Following error is getting generated when trying to load the model.

ERROR:

TypeError Traceback (most recent call last) in () 8 detector.setModelTypeAsRetinaNet() 9 detector.setModelPath( os.path.join(execution_path , "resnet50_coco_best_v2.0.1.h5")) ---> 10 detector.loadModel() 11 detections = detector.detectObjectsFromImage(input_image=os.path.join(execution_path , "image.jpg"), output_image_path=os.path.join(execution_path , "imagenew.jpg")) 12

~\Anaconda3\envs\Data_Injest\lib\site-packages\imageai\Detectioninit.py in loadModel(self, detection_speed) 183 elif(self.modelType == "retinanet"): 184 model = resnet50_retinanet(num_classes=80) --> 185 model.load_weights(self.modelPath) 186 self.__model_collection.append(model) 187 self.modelLoaded = True

~\Anaconda3\envs\Data_Injest\lib\site-packages\keras\engine\network.py in load_weights(self, filepath, by_name, skip_mismatch, reshape) 1164 else: 1165 saving.load_weights_from_hdf5_group( -> 1166 f, self.layers, reshape=reshape) 1167 1168 def _updated_config(self):

~\Anaconda3\envs\Data_Injest\lib\site-packages\keras\engine\saving.py in load_weights_from_hdf5_group(f, layers, reshape) 1056 ' elements.') 1057 weight_value_tuples += zip(symbolic_weights, weight_values) -> 1058 K.batch_set_value(weight_value_tuples) 1059 1060

~\Anaconda3\envs\Data_Injest\lib\site-packages\keras\backend\tensorflow_backend.py in batch_set_value(tuples) 2468 assign_ops.append(assign_op) 2469 feed_dict[assign_placeholder] = value -> 2470 get_session().run(assign_ops, feed_dict=feed_dict) 2471 2472

~\Anaconda3\envs\Data_Injest\lib\site-packages\tensorflow\python\client\session.py in run(self, fetches, feed_dict, options, run_metadata) 927 try: 928 result = self._run(None, fetches, feed_dict, options_ptr, --> 929 run_metadata_ptr) 930 if run_metadata: 931 proto_data = tf_session.TF_GetBuffer(run_metadata_ptr)

~\Anaconda3\envs\Data_Injest\lib\site-packages\tensorflow\python\client\session.py in _run(self, handle, fetches, feed_dict, options, run_metadata) 1150 if final_fetches or final_targets or (handle and feed_dict_tensor): 1151 results = self._do_run(handle, final_targets, final_fetches, -> 1152 feed_dict_tensor, options, run_metadata) 1153 else: 1154 results = []

~\Anaconda3\envs\Data_Injest\lib\site-packages\tensorflow\python\client\session.py in _do_run(self, handle, target_list, fetch_list, feed_dict, options, run_metadata) 1326 if handle is None: 1327 return self._do_call(_run_fn, feeds, fetches, targets, options, -> 1328 run_metadata) 1329 else: 1330 return self._do_call(_prun_fn, handle, feeds, fetches)

~\Anaconda3\envs\Data_Injest\lib\site-packages\tensorflow\python\client\session.py in _do_call(self, fn, args) 1332 def _do_call(self, fn, args): 1333 try: -> 1334 return fn(*args) 1335 except errors.OpError as e: 1336 message = compat.as_text(e.message)

~\Anaconda3\envs\Data_Injest\lib\site-packages\tensorflow\python\client\session.py in _run_fn(feed_dict, fetch_list, target_list, options, run_metadata) 1317 self._extend_graph() 1318 return self._call_tf_sessionrun( -> 1319 options, feed_dict, fetch_list, target_list, run_metadata) 1320 1321 def _prun_fn(handle, feed_dict, fetch_list):

~\Anaconda3\envs\Data_Injest\lib\site-packages\tensorflow\python\client\session.py in _call_tf_sessionrun(self, options, feed_dict, fetch_list, target_list, run_metadata) 1405 return tf_session.TF_SessionRun_wrapper( 1406 self._session, options, feed_dict, fetch_list, target_list, -> 1407 run_metadata) 1408 1409 def _call_tf_sessionprun(self, handle, feed_dict, fetch_list):

TypeError: TF_SessionRun_wrapper: expected all values in input dict to be ndarray

OlafenwaMoses commented 5 years ago

Kindly update to the latest version of ImageAI

pip3 instal imageai --upgrade

patel-priyankkumar commented 3 years ago

still not working