drone-dataset's Introduction

Hi 👋, I'm Mehdi

Machine Learning Engineer

🔭 I’m currently working on Virtual Try-On Systems
🔭 I was working on Vision & AI on Drones
📝 I have experiences on AI in embedded systems like STM and Nvidia Jetson series.
💬 I generally use Tensorflow / PyTorch / OpenCV / Pandas / Sklearn etc.
📫 How to reach me: [email protected]

Languages and Tools:

drone-dataset's People

Contributors

Stargazers

Watchers

drone-dataset's Issues

problems while training with dataset

Hey
Thanks for your dataset.
Unfortunately I have some issues..
I am working with tensorflow and took your Images as dataset.

When I want to try the model I get the following error:

flu@flubuntu:~/Tensorflow/scripts/preprocessing$ python3 generate_tfrecord.py --label=drone --csv_input=/home/flu/Tensorflow/workspace/training_demo/annotations/test_labels.csv --img_path=/home/flu/Tensorflow/workspace/training_demo/images/test --output_path=/home/flu/Tensorflow/workspace/training_demo/annotations/test.record
/home/flu/.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:523: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_qint8 = np.dtype([("qint8", np.int8, 1)])
/home/flu/.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:524: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_quint8 = np.dtype([("quint8", np.uint8, 1)])
/home/flu/.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:525: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_qint16 = np.dtype([("qint16", np.int16, 1)])
/home/flu/.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:526: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_quint16 = np.dtype([("quint16", np.uint16, 1)])
/home/flu/.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:527: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  _np_qint32 = np.dtype([("qint32", np.int32, 1)])
/home/flu/.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py:532: FutureWarning: Passing (type, 1) or '1type' as a synonym of type is deprecated; in a future version of numpy, it will be understood as (type, (1,)) / '(1,)type'.
  np_resource = np.dtype([("resource", np.ubyte, 1)])
Traceback (most recent call last):
  File "generate_tfrecord.py", line 117, in <module>
    tf.app.run()
  File "/home/flu/.local/lib/python3.6/site-packages/tensorflow/python/platform/app.py", line 125, in run
    _sys.exit(main(argv))
  File "generate_tfrecord.py", line 108, in main
    tf_example = create_tf_example(group, path)
  File "generate_tfrecord.py", line 64, in create_tf_example
    image = Image.open(encoded_jpg_io)
  File "/usr/lib/python3/dist-packages/PIL/Image.py", line 2590, in open
    % (filename if filename else fp))
OSError: cannot identify image file <_io.BytesIO object at 0x7fb7f8d869e8>

Can you help me with that?

Thx so much