along/semantic-segmentation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改参数

Browse Source
This commit is contained in:
weixin_46229132 2025-07-18 09:02:50 +08:00
parent 8148a58584
commit 889091a4af
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train_JL/train_JL.py
View File

@ -21,7 +21,7 @@ def parse_args():
parser = argparse.ArgumentParser(description="pytorch deeplabv3 training")
parser.add_argument(
"--data-path", default=r"E:\datasets\wlk_right_448", help="VOCdevkit root")
"--data-path", default=r"E:\RSdata\wlk_right_448", help="VOCdevkit root")
parser.add_argument("--num-classes", default=7, type=int)
parser.add_argument("--device", default="cuda", help="training device")
parser.add_argument("--batch-size", default=4, type=int)

2
train_JL_jpg/geotiff_utils.py
View File

@ -166,7 +166,7 @@ class VOCYJSSegmentation(SegmentationDataset):
def __getitem__(self, index):
img_name = self.image_list[index].split('.')[0]+'.jpg'
mask_name = self.image_list[index].split('.')[0]+'.png'
mask_name = mask_name.replace('img', 'mask')
# mask_name = mask_name.replace('img', 'mask')
img_LS = np.array(Image.open(os.path.join(
self._image_LS_dir, img_name))).astype(np.float32)
mask = np.array(Image.open(os.path.join(

2
train_JL_jpg/train_JL.py
View File

@ -25,7 +25,7 @@ def parse_args():
parser.add_argument("--num-classes", default=7, type=int)
parser.add_argument("--device", default="cuda", help="training device")
parser.add_argument("--batch-size", default=4, type=int)
parser.add_argument("--epochs", default=50, type=int, metavar="N",
parser.add_argument("--epochs", default=200, type=int, metavar="N",
help="number of total epochs to train")
parser.add_argument('--lr', default=0.005, type=float,
help='initial learning rate')

2
train_LS/train_LS.py
View File

@ -25,7 +25,7 @@ def parse_args():
parser.add_argument("--num-classes", default=13, type=int)
parser.add_argument("--device", default="cuda", help="training device")
parser.add_argument("--batch-size", default=8, type=int)
parser.add_argument("--epochs", default=50, type=int, metavar="N",
parser.add_argument("--epochs", default=200, type=int, metavar="N",
help="number of total epochs to train")
parser.add_argument('--lr', default=0.005, type=float,
help='initial learning rate')