-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathinference.py
20 lines (18 loc) · 863 Bytes
/
inference.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from utils.operation import *
from utils.file import *
import argparse
if __name__ == '__main__':
print("start model arts train.")
parser = argparse.ArgumentParser()
parser.add_argument('--data_url', required=False, type=str, help='the training and validation data path')
parser.add_argument('--train_url', required=False, type=str, help='the path to save training outputs')
parser.add_argument('--init_method', default='', type=str, help='the training output results on local')
opt = parser.parse_args()
print(opt)
test1(
in_channels=3, out_channels=8, net_name="deeplabv3",
weights_path=r"D:\AI\project\data\weights\ckpt_1_32.78.pth",
test_image_root=r"D:\AI\project\data\baidu_lane_line\TestSet\ColorImage",
batch_size=2, resize=(512, 512), crop_offset=(730, 0),
pretrained=False
)