From d0f86018f1047addc5f614d415d0c7bc048c3315 Mon Sep 17 00:00:00 2001
From: unanmed <1319491857@qq.com>
Date: Thu, 12 Mar 2026 20:41:03 +0800
Subject: [PATCH] =?UTF-8?q?chore:=20=E8=B0=83=E6=95=B4=E9=83=A8=E5=88=86?=
 =?UTF-8?q?=E4=BE=9D=E8=B5=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ginka/dataset.py       | 6 ------
 ginka/maskGIT/mask.py  | 1 -
 ginka/train_maskGIT.py | 4 ++--
 requirements.txt       | 6 +++---
 train.txt              | 1 -
 5 files changed, 5 insertions(+), 13 deletions(-)
 delete mode 100644 train.txt

diff --git a/ginka/dataset.py b/ginka/dataset.py
index 512d0a2..9fcf55d 100644
--- a/ginka/dataset.py
+++ b/ginka/dataset.py
@@ -1,12 +1,6 @@
 import json
-import math
-import random
 import torch
-import torch.nn.functional as F
 from torch.utils.data import Dataset
-import torch
-import torch.nn.functional as F
-from typing import List
 
 def load_data(path: str):
     with open(path, 'r', encoding="utf-8") as f:
diff --git a/ginka/maskGIT/mask.py b/ginka/maskGIT/mask.py
index 0cfb31a..72a78ae 100644
--- a/ginka/maskGIT/mask.py
+++ b/ginka/maskGIT/mask.py
@@ -1,5 +1,4 @@
 import random
-import torch
 import numpy as np
 from scipy.ndimage import binary_dilation, binary_erosion
 
diff --git a/ginka/train_maskGIT.py b/ginka/train_maskGIT.py
index 8722fd9..9faa29d 100644
--- a/ginka/train_maskGIT.py
+++ b/ginka/train_maskGIT.py
@@ -9,8 +9,8 @@ import torch.nn.functional as F
 import torch.optim as optim
 import cv2
 import numpy as np
-from torch_geometric.loader import DataLoader
 from tqdm import tqdm
+from torch.utils.data import DataLoader
 from .maskGIT.model import GinkaMaskGIT
 from .dataset import GinkaMaskGITDataset
 from shared.image import matrix_to_image_cv
@@ -61,7 +61,7 @@ disable_tqdm = not sys.stdout.isatty()
 def parse_arguments():
     parser = argparse.ArgumentParser(description="training codes")
     parser.add_argument("--resume", type=bool, default=False)
-    parser.add_argument("--state_ginka", type=str, default="result/vae/ginka-100.pth")
+    parser.add_argument("--state_ginka", type=str, default="result/transformer/ginka-100.pth")
     parser.add_argument("--train", type=str, default="ginka-dataset.json")
     parser.add_argument("--validate", type=str, default="ginka-eval.json")
     parser.add_argument("--epochs", type=int, default=100)
diff --git a/requirements.txt b/requirements.txt
index aa7eed8..c451531 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 torch
-torchvision
-torchaudio
 tqdm
 torch-geometric
 transformers
-torch-scatter
\ No newline at end of file
+scipy
+numpy
+cv2
\ No newline at end of file
diff --git a/train.txt b/train.txt
deleted file mode 100644
index fba325c..0000000
--- a/train.txt
+++ /dev/null
@@ -1 +0,0 @@
-python3 -u -m ginka.train_wgan --epochs 200 --checkpoint 20 --resume true --state_ginka result/wgan/ginka-400.pth --state_minamo result/wgan/minamo-400.pth >> output.log
\ No newline at end of file