auto labeler

Labeler otomatis

Auto Labeler adalah alat anotasi gambar otomatis yang memanfaatkan kekuatan API Visi GPT-4 Openai untuk mendeteksi objek dalam gambar dan memberikan anotasi kotak pembatas.

Fitur

• Deteksi Objek: Secara otomatis mengidentifikasi objek dalam gambar.
• Anotasi kotak pembatas: Menghasilkan kotak pembatas di sekitar objek yang terdeteksi.
• Berbagai format: mendukung menghemat anotasi di kedua deteksi objek TensorFlow dan format JSON.
• Label Kustom: Memungkinkan untuk menentukan label objek khusus untuk dideteksi.

Memulai

Prasyarat

• Python 3.6 atau lebih tinggi

Instalasi

Klon Repositori ke Mesin Lokal Anda:

git clone https://github.com/Flode-Labs/auto-labeler.git

Penggunaan

1. Instal Paket yang Diperlukan:

 pip install -r requirements.txt

2. Atur tombol API OpenAI Anda:

 api_key = "YOUR_API_KEY"

3. Tentukan jalur folder dengan gambar yang ingin Anda beri label:

 input_folder = '/path/to/input/folder'

4. Tentukan jalur folder tempat Anda ingin menyimpan label:

 output_folder = '/path/to/output/folder'

5. Tentukan format di mana Anda ingin menyimpan label:

 output_format = 'json' # or 'tf'

6. Tentukan daftar label yang ingin Anda deteksi:

 labels = [ 'cat' , 'dog' , 'bird' ]

7. Jalankan skrip utama:

python main.py

Fungsi

• encode_image(image_path) : mengkodekan format gambar ke base64.
• detect_objects(image_path, labels) : Mendeteksi objek dalam gambar sebagaimana ditentukan oleh label.
• save_tf_annotations(...) : Menyimpan anotasi dalam format deteksi objek TensorFlow XML.
• save_json_annotations(...) : Menyimpan anotasi dalam format JSON.
• process_images_in_folder(...) : memproses beberapa gambar dalam folder yang ditentukan.

Ucapan Terima Kasih

• Openai untuk menyediakan API Visi GPT-4.
• Komunitas Python untuk memelihara dan meningkatkan perpustakaan yang digunakan dalam proyek ini.

---

Pastikan Anda mengganti YOUR_API_KEY dengan kunci API OpenAI Anda yang sebenarnya.