Dự án GatedConvolution là một phiên bản tái hiện của mô hình sửa chữa hình ảnh trong bài viết khoa học về Sửa Chữa Hình Ảnh Miễn Phí Hình Dạng với Hợp Thành Gated Convolution. Địa chỉ dự án: GatedConvolution

Cấu trúc Thư Mục và Giới Thiệu

GatedConvolution/
├── config/
│   ├── inpaint.yml
│   └── ...
├── examples/
│   ├── input.png
│   ├── mask.png
│   └── output.png
├── util/
│   ├── data_from_fnames.py
│   ├── mask_from_fnames.py
│   └── ...
├── inpaint_model.py
├── inpaint_model_gc.py
├── inpaint_ops.py
├── LICENSE
├── README.md
├── test.py
├── train.py
└── trainer.py

Giới Thiệu Cấu Trúc Thư Mục

• config/: Chứa các tệp cấu hình như inpaint.yml, dùng để thiết lập các tham số cho quá trình huấn luyện và kiểm thử.
• examples/: Lưu trữ các tệp hình ảnh mẫu bao gồm hình ảnh đầu vào, mặt nạ và hình ảnh đầu ra.
• util/: Chứa các script tiện ích như data_from_fnames.py và mask_from_fnames.py, dùng để xử lý dữ liệu.
• inpaint_model.py: Định nghĩa cấu trúc chính của mô hình sửa chữa hình ảnh.
• inpaint_model_gc.py: Định nghĩa mô hình sửa chữa hình ảnh sử dụng hợp thành gated convolution.
• inpaint_ops.py: Định nghĩa các phép toán sửa chữa hình ảnh.
• LICENSE: Tệp giấy phép mở nguồn của dự án.
• README.md: Tệp mô tả dự án.
• test.py: Script dùng để kiểm thử mô hình.
• train.py: Script dùng để huấn luyện mô hình.
• trainer.py: Script thực hiện quá trình huấn luyện.

Giới Thiệu Các Tập Tin Khởi Động Dự Án

train.py

train.py là một trong những tập tin khởi động của dự án, dùng để bắt đầu quá trình huấn luyện. Dưới đây là các chức năng chính của tập tin:

• Huấn luyện mô hình: Bắt đầu quá trình huấn luyện mô hình bằng cách đọc các tham số từ tệp cấu hình inpaint.yml.
• Tái khởi động huấn luyện: Hỗ trợ tái khởi động huấn luyện từ trạng thái trước đó thông qua việc thiết lập tham số MODEL_RESTORE.

test.py

test.py là tập tin khởi động khác, dùng để kiểm thử mô hình đã được huấn luyện. Dưới đây là các chức năng chính của tập tin:

• Kiểm thử mô hình: Kiểm tra hiệu suất của mô hình bằng cách chỉ định hình ảnh đầu vào, mặt nạ và đường dẫn đầu ra.
• Tạo hình ảnh đã sửa chữa: Tạo và lưu trữ hình ảnh đã sửa chữa tại đường dẫn được chỉ định.

Giới Thiệu Tập Tin Cấu Hình Dự Án

config/inpaint.yml

inpaint.yml là tệp cấu hình chính của dự án, dùng để thiết lập các tham số cho quá trình huấn luyện và kiểm thử. Dưới đây là các mục cấu hình chính:

• DATA_FLIST: Đường dẫn tới tệp danh sách các tệp dữ liệu huấn luyện.
• LOG_DIR: Đường dẫn lưu trữ các tệp nhật ký.
• IMG_SHAPES: Kích thước của hình ảnh đầu vào.
• MODEL_RESTORE: Đường dẫn tới mô hình được sử dụng khi tái khởi động huấn luyện.

Ví Dụ Cấu Hình

DATA_FLIST: "đường/dẫn/tới/data_flist.txt"
LOG_DIR: "đường/dẫn/tới/log_dir"
IMG_SHAPES: [256, 256, 3]
MODEL_RESTORE: "20180115220926508503_places2_model"

Bằng cách chỉnh sửa các mục cấu hình này, bạn có thể tùy chỉnh hành vi của quá trình huấn luyện và kiểm thử.