1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning cho Ảnh với PyTorch
Connected

Bài tập

Tạo ảnh

Giờ bạn đã thiết kế và huấn luyện xong GAN, đã đến lúc đánh giá chất lượng ảnh mà mô hình có thể tạo ra. Trước hết, bạn sẽ quan sát trực quan để xem các ảnh sinh ra có giống Pokemon hay không. Để làm điều này, bạn sẽ tạo nhiễu ngẫu nhiên làm đầu vào cho generator, truyền nó qua mô hình và vẽ các đầu ra.

Deep Convolutional Generator với trọng số đã được huấn luyện có sẵn dưới tên gen. torch và matplotlib.pyplot với bí danh plt đã được nhập sẵn cho bạn.

Hướng dẫn

100 XP
  • Tạo một tensor nhiễu ngẫu nhiên có dạng num_images_to_generate theo 16 (kích thước nhiễu đầu vào bạn dùng để huấn luyện generator) và gán vào noise.
  • Tạo ảnh bằng cách truyền nhiễu vào generator và gán kết quả vào fake.
  • Bên trong vòng lặp for, cắt fake để lấy ảnh thứ i và gán vào image_tensor.
  • Hoán vị các chiều của image_tensor từ (màu, cao, rộng) sang (cao, rộng, màu) và gán đầu ra vào image_tensor_permuted.