From 1326c7bbc3710971adaace97d6e9781585601d66 Mon Sep 17 00:00:00 2001
From: linaom1214 <linaom1214@gmail.com>
Date: Mon, 16 May 2022 09:25:21 +0800
Subject: [PATCH] fix

---
 main.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/main.py b/main.py
index b13090f..fac3ee5 100644
--- a/main.py
+++ b/main.py
@@ -4,7 +4,6 @@
 import os
 from datetime import datetime
 from copy import deepcopy
-from torchmetrics import AUC
 from tqdm import tqdm
 from skimage.segmentation import mark_boundaries
 import matplotlib.pyplot as plt
@@ -20,7 +19,6 @@
 from torchvision import transforms
 from torchvision.transforms.functional import to_pil_image
 from torchvision.utils import save_image
-from zmq import device
 
 from net.Unet import Unet
 from net.FCN import fcn