Artist-Style-Transfer-CycleGAN

Artist Style Transfer CycleGAN

Ngôn ngữ lập trình

Công nghệ sử dụng

Project Structure

Mở rộng

``` ├───assets ├───checkpoints │ ├───ce │ ├───mo │ ├───uk │ └───vg ├───datasets │ ├───cezanne2photo │ │ ├───testA │ │ ├───testB │ │ ├───trainA │ │ │ └───A │ │ └───trainB │ │ └───B │ ├───monet2photo │ │ ├───testA │ │ ├───testB │ │ ├───trainA │ │ │ └───A │ │ └───trainB │ │ └───B │ ├───ukiyoe2photo │ │ ├───testA │ │ ├───testB │ │ ├───trainA │ │ │ └───A │ │ └───trainB │ │ └───B │ └───vangogh2photo │ ├───testA │ ├───testB │ ├───trainA │ │ └───A │ └───trainB │ └───B ├───images └───out ```

Datasets

Download vào thư mục datasets\:

• Monet
• Cezanne
• Ukiyo-e
• Vangogh

Lưu ý: sau khi download và giải nén, phải sửa cấu trúc thư mục theo định dạng:

datasets
└───vangogh2photo
    ├───testA
    ├───testB
    ├───trainA
    │   └───A
    └───trainB
        └───B

Results

Loss chart

Biểu đồ Loss của model trong quá trình train 5000 epoches

Test

Test với pre-train

Với mỗi style ảnh (Cezanne, Monet, Ukio-e, Van Gogh), nhóm đem test với 4 ảnh và thu được kết quả như sau:

Test với model tự custom

Danh sách các ảnh Input

Cứ 50 epoches trong tổng số 5000 epoches, nhóm sẽ lấy ra 1 checkpoint để test. Thu được kết quả như sau:

• Kết quả Test với ảnh Test 1 ứng với các epoch 0 – 5000 (step: 50):

• Kết quả Test với ảnh Test 2 ứng với các epoch 0 – 5000 (step: 50):

• Kết quả Test với ảnh Test 3 ứng với các epoch 3000 – 4000 (step: 10):

• Kết quả Test với ảnh Test 4 ứng với các epoch 3000 – 4000 (step: 10):

Report

Xem report tại: Report - password: hcmute

References

• Rahul Bhalley and Jianlin Su
• https://github.com/rahulbhalley/cyclegan-qp

This site is open source. Improve this page.