zenloger/autopilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ref: use uploaded weights

Browse Source
This commit is contained in:
russian_proger
2026-05-31 13:07:08 +03:00
parent 5f1594466b
commit ec53e9ec8d
2 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gan.py
View File

@@ -16,7 +16,7 @@ from vision_chunk import VisionChunk
ROOT_DIR = Path(__file__).resolve().parent
MODEL_FILE = ROOT_DIR / "models" / "GAN" / "src" / "model.py"
DEFAULT_CHECKPOINT_PATH = ROOT_DIR / "models" / "GAN" / "runs" / "checkpoints" / "best.pth"
DEFAULT_CHECKPOINT_PATH = ROOT_DIR / "models" / "TrainedWeights" / "GAN.pth"
IMAGE_SIZE = (256, 256)
CHECKPOINT_ENV = "GAN_CHECKPOINT"

7
sian_similarity.py
View File

@@ -17,11 +17,8 @@ MODEL_FILE = ROOT_DIR / "models" / "SiaN-similarity" / "model.py"
DEFAULT_CHECKPOINT_PATH = (
ROOT_DIR
/ "models"
/ "SiaN-similarity"
/ "runs"
/ "gan_training"
/ "checkpoints"
/ "best_model.pt"
/ "TrainedWeights"
/ "SiaN-Similarity.pt"
)
IMAGE_SIZE = (256, 256)