Python thêm nhãn vào hình ảnh

Các nhãn tôi đã sử dụng

Cho đến nay tôi mới chỉ sử dụng Tinder cho dự án này, nhưng tôi nhanh chóng nhận ra rằng có rất nhiều hình ảnh trên Tinder và nhiều hình ảnh trong số đó khá kỳ lạ. Động vật, nhóm, hoàng hôn, bạn có thể tìm thấy mọi thứ trên Tinder

Do đó, tôi quyết định không chỉ sử dụng các nhãn thích và không thích, vì sẽ rất khó (đối với Mạng thần kinh và đối với tôi) để so sánh hình ảnh cơ thể với hình ảnh khuôn mặt với cảnh hoàng hôn, v.v.

Để giải quyết vấn đề này, tôi đã thử sử dụng các nhãn

• body_like
• body_dislike
• face_like
• face_dislike
• vô tư

Bằng cách này, chúng ta có thể so sánh các hình ảnh hơi giống nhau với nhau. Ngoài ra, danh mục thờ ơ là điều cần thiết vì rõ ràng là các cô gái xinh đẹp cũng sử dụng hình ảnh động vật và tôi không muốn một bức ảnh không thực sự có thể phân loại được để ảnh hưởng đến quyết định của bot của tôi

Thiết lập dự án

Tiếp tục sau khi tải xuống các hình ảnh mong muốn, cấu trúc dự án của chúng ta sẽ giống như thế này

Bây giờ, trong thư mục dữ liệu của chúng tôi, chúng tôi sẽ thêm các thư mục cho tất cả các danh mục ghi nhãn của chúng tôi

Các hình ảnh sẽ được gắn nhãn hiện có trong thư mục tải xuống

Đầu tiên chúng tôi cài đặt matplotlib

				
					pip install matplotlib

				
			

Kịch bản tôi đã sử dụng

Để dán nhãn cho hình ảnh, tôi đã sử dụng đoạn script sau

				
					import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import os


def save_image(like_value, photo, img):
   filename = 'data/'

   # body like
   if like_value == 'b':
       print('body like')
       filename += 'body_like/' + photo
       plt.imsave(filename, img)

   # body dislike
   if like_value == 'bb':
       print('body dislike')
       filename += 'body_dislike/' + photo
       plt.imsave(filename, img)

   # face like
   if like_value == 'f':
       print('face like')
       filename += 'face_like/' + photo
       plt.imsave(filename, img)

   # face dislike
   if like_value == 'ff':
       print('face dislike')
       filename += 'face_dislike/' + photo
       plt.imsave(filename, img)

   # indifferent
   if like_value == 'i':
       print('indifferent')
       filename += 'indifferent/' + photo
       plt.imsave(filename, img)


def main():
   base_path = 'data/downloads/'
   for photo in os.listdir(base_path):
       photo_path = base_path + photo
       img = mpimg.imread(photo_path)
       plt.imshow(img)
       plt.show()

       like_value = input()

       # save pic to new folder and delete it from 'to_label' folder
       save_image(like_value, photo, img)
       os.remove(photo_path)


if __name__ == "__main__":
   main()


				
			

Tập lệnh này sẽ hiển thị hết hình ảnh này đến hình ảnh khác trong thư mục tải xuống và đợi người dùng nhập liệu. ví dụ: nếu bạn nhập 'b' và nhấn enter trong thiết bị đầu cuối của mình, hình ảnh sẽ bị xóa khỏi thư mục tải xuống và được lưu vào thư mục body_like

• b -> body_like
• bb -> body_dislike
• f -> face_like
• ff -> face_dislike
• tôi -> thờ ơ

thận trọng. nếu bạn nhấn bất kỳ phím nào khác, hình ảnh sẽ bị xóa và không được lưu vào bất kỳ thư mục nào khác

Phần kết luận

Bài viết này trình bày cách sử dụng Python để gắn nhãn hình ảnh để chuẩn bị cho việc đào tạo một mô hình máy học

Chúng tôi đã sử dụng một tập lệnh Python đơn giản để xóa hình ảnh khỏi một thư mục và đặt chúng vào một thư mục mới theo nhãn của chúng

Trong bài đăng tiếp theo, chúng ta sẽ xem xét cách chúng ta có thể sử dụng các hình ảnh được dán nhãn. Do đó, chúng tôi sẽ đào tạo Mạng nơ-ron để có thể gán nhãn cho hình ảnh mới theo sở thích của chúng tôi