Xây dựng U-Net: định nghĩa các tầng

Trong bài này và bài tiếp theo, bạn sẽ tự xây dựng kiến trúc U-Net từ đầu. Ở đây, bạn bắt đầu bằng cách định nghĩa phương thức __init__() nơi bạn sẽ khai báo các tầng và các khối tầng để mô hình sử dụng.

Encoder và các phép tích chập chuyển vị đã được định nghĩa sẵn cho bạn. Phần còn lại là các khối tích chập của decoder. Bạn cần truyền số kênh đầu vào và đầu ra phù hợp cho từng khối, có tính đến các skip connection.

Khối đầu tiên, dec1, sẽ nhận đầu vào là phép nối giữa đầu ra của upconv3 với đầu ra của enc3. Đầu ra của dec1, đến lượt nó, phải bằng với đầu ra của enc3. Bạn có thể điền tất cả các kích thước đầu vào và đầu ra còn thiếu không?

Định nghĩa ba khối tích chập trong decoder bằng cách truyền số kênh đầu vào và đầu ra phù hợp cho từng khối.

Bài tập

Xây dựng U-Net: định nghĩa các tầng

Hướng dẫn

.css-6su6fj{-webkit-flex-shrink:0;-ms-flex-negative:0;flex-shrink:0;}Bài tập

Hướng dẫn

Bài tập