CODEWITHSUNDEEP
pythontensorflowimageai
Philipp
Philipp

Reputation: 662

Using ImageAI I get => AttributeError: module 'tensorflow' has no attribute 'to_float'

I want to detect objects in images, following this tutorial (https://medium.com/deepquestai/train-object-detection-ai-with-6-lines-of-code-6d087063f6ff). However I receive an error message, which I cannot resolve. What can I do about it, given that I cannot change the source code from imageai and therefore not fix the error this way (https://github.com/google/tangent/issues/95)?

These are my imports:

!pip3 install tensorflow-gpu==1.13.1
!pip install imageai --upgrade
from imageai.Detection.Custom import DetectionModelTrainer

I run this code:

data_path = 'leaf-images-with-pascal-voc-annotations/'

trainer = DetectionModelTrainer()
trainer.setModelTypeAsYOLOv3()
trainer.setDataDirectory(data_directory=data_path)
trainer.setTrainConfig(object_names_array=['leaf'], batch_size=16, num_experiments=100, 
                       train_from_pretrained_model="pretrained-yolov3.h5")
trainer.trainModel()

I tried using different versions of tensorflow But receive this error message:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-28-d42b2127d681> in <module>
      6 trainer.setTrainConfig(object_names_array=['leaf'], batch_size=16, num_experiments=100, 
      7                        train_from_pretrained_model="pretrained-yolov3.h5")
----> 8 trainer.trainModel()

/opt/conda/lib/python3.6/site-packages/imageai/Detection/Custom/__init__.py in trainModel(self)
    272             noobj_scale=self.__train_noobj_scale,
    273             xywh_scale=self.__train_xywh_scale,
--> 274             class_scale=self.__train_class_scale,
    275         )
    276 

/opt/conda/lib/python3.6/site-packages/imageai/Detection/Custom/__init__.py in _create_model(self, nb_class, anchors, max_box_per_image, max_grid, batch_size, warmup_batches, ignore_thresh, multi_gpu, lr, grid_scales, obj_scale, noobj_scale, xywh_scale, class_scale)
    551                     noobj_scale=noobj_scale,
    552                     xywh_scale=xywh_scale,
--> 553                     class_scale=class_scale
    554                 )
    555         else:

/opt/conda/lib/python3.6/site-packages/imageai/Detection/Custom/yolo.py in create_yolov3_model(nb_class, anchors, max_box_per_image, max_grid, batch_size, warmup_batches, ignore_thresh, grid_scales, obj_scale, noobj_scale, xywh_scale, class_scale)
    292                             noobj_scale,
    293                             xywh_scale,
--> 294                             class_scale)([input_image, pred_yolo_1, true_yolo_1, true_boxes])
    295 
    296     # Layer 83 => 86

/opt/conda/lib/python3.6/site-packages/imageai/Detection/Custom/yolo.py in __init__(self, anchors, max_grid, batch_size, warmup_batches, ignore_thresh, grid_scale, obj_scale, noobj_scale, xywh_scale, class_scale, **kwargs)
     22         max_grid_h, max_grid_w = max_grid
     23 
---> 24         cell_x = tf.to_float(tf.reshape(tf.tile(tf.range(max_grid_w), [max_grid_h]), (1, max_grid_h, max_grid_w, 1, 1)))
     25         cell_y = tf.transpose(cell_x, (0,2,1,3,4))
     26         self.cell_grid = tf.tile(tf.concat([cell_x,cell_y],-1), [batch_size, 1, 1, 3, 1])

AttributeError: module 'tensorflow' has no attribute 'to_float'

Upvotes: 3

Views: 6627

Answers (2)

prem
prem

Reputation: 3538

I was also getting the same error while trying to use the to_float method in the version 2.3.0

It seems like, this method has been removed in the newer versions of the library.

To get it working, I have changed my code to use the cast method instead of to_float.

Below is the sample code which worked for me

num=5
#as_float = tf.to_float(num)
#Change the above code line and use cast method instead
as_float=tf.cast(num, tf.float32)
as_float

Upvotes: 4

Philipp
Philipp

Reputation: 662

There appears to be a currently unfixed state of the ImageAI library where it is not compatible with the newest versions of tensorflow, etc.

Using these versions worked for me:

#Currently I found these to work together:
pip install opencv-python==4.1.2.30
pip install keras==2.3.1
pip install tensorflow==1.14.0
pip install tensorflow-gpu==1.14.0
pip install imageai --upgrade
NOTE: using imageai == 2.1.5

Upvotes: 2

Related Questions