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

Connected

Bài tập

Fréchet Inception Distance

Việc quan sát trực quan các ảnh được tạo là một khởi đầu tốt. Nhưng khi ảnh trông ổn, bạn sẽ cần đánh giá định lượng chính xác hơn để hiểu hiệu năng của generator. Bạn sẽ đánh giá GAN của mình bằng Fréchet Inception Distance (FID).

Hai tensor chứa ảnh giả và ảnh thật, mỗi loại 32 ví dụ, đã được cung cấp lần lượt dưới dạng fake và real. Hãy dùng chúng để tính FID!

Hướng dẫn

100 XP
  • Import FrechetInceptionDistance từ module torchmetrics phù hợp.
  • Khởi tạo metric FID dựa trên tầng đặc trưng Inception thứ 64 và gán vào fid.
  • Cập nhật fid với tensor ảnh thật, nhân với 255 và ép kiểu sang torch.uint8.
  • Tính metric fid, gán đầu ra vào fid_score.