Skip to content

phamkhai108/Image-classification-api

Repository files navigation

Dự án phân loại động vật có vú qua API với 5 loài 'Cat', 'Dog', 'Rose', 'Elephant', 'Horse', 'Lion'

chức năng :

  1. Dư đoán và trả về kết quả thông qua API.
    • Đầu vào: các trường, trong đó có trường hình ảnh
    • Đàu ra: kết quả dự đoán, gồm có tên đọngo vật và độ tin cậy mà mô hình dự đoán ra loài động vật đó.
  2. Xem kết quả khi các giá trị đã up lên database
  3. Xóa các đối tượng trong database thông qua id của chúng

Tổng quan thực hiện project:

  1. Thu thập dữ liệu.
  2. Train model phân loại 5 loài động vật với kiến trúc CNN dùng framework 'Tensorflow'. Chi tiết link (data và googlecolab): Data and googlecolab
  3. Thực hiện xây dựng Api với Django-rest-framework.

công nghệ sử dụng

  1. Ngôn ngữ lập trình: python
  2. Framework và thư viện chính: Django, tensorflow, Pillow, numpy
    • Django: tạo khung viết api.

      Link document: Django-rest-framewor

    • tensorflow: tạo dựng và train model.

      Link tìm hiểu và học tập: Tensorflow

    • Pillow: dùng để lấy ảnh đầu vào từ request

      Link tham khảo: Pillow

    • numpy: xử lý ảnh và chuyển ảnh về mảng phù hợp với việc dùng model dự đoán và đưa ra kết quả

      Link tham khảo: Numpy

Hướng dẫn cách chạy project.

  1. Tải về project bằng câu lệnh 'git clone https://github.com/phamkhai108/Image-classification-api.git'
  2. Setup và cài đặt các thư viện cần thiết.
    • python version 3.11.6
    • Framework và thư viện: chạy lệnh 'pip install requirements.txt' để cài đặt các thư viện và framework cần thiết
  3. Tiến hành chạy project.
    • Chuyển đến đường dẫn project chay câu lệnh 'py manage.py runserver 8100' ('8100' là cổng có thể thay đổi, nếu không có giá trị đó mặc định sẽ chạy trên cổng '8000')
    • khi chạy project chúng ta có các đường dẫn để thử như sau, mọi người có thể dùng postman để thử.
    • Gồm 3 đường dẫn:

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages