개정판 28822594
issue #1366: add comment
Change-Id: I5f82e013c9601d24a38a6552eb6b4cdec838eaab
| DTI_PID/WebServer/symbol_recognition/test_doftech_all_images.py | ||
|---|---|---|
| 56 | 56 |
parser.add_argument("--nms_threshold", type=float, default=0.6)
|
| 57 | 57 |
parser.add_argument("--pre_trained_model_type", type=str, choices=["model", "params"], default="params")
|
| 58 | 58 |
parser.add_argument("--pre_trained_model_path", type=str, default=trained_model1)
|
| 59 |
parser.add_argument("--pre_trained_model_path2", type=str, default=trained_model2)
|
|
| 59 |
#parser.add_argument("--pre_trained_model_path2", type=str, default=trained_model2)
|
|
| 60 | 60 |
parser.add_argument("--data_path_test", type=str, default="E:\Projects\DTIPID_GERRIT\DTI_PID\WebServer\symbol_recognition\pid_images/", help="the root folder of dataset") #Doftech VOCdevkit
|
| 61 | 61 |
opt = parser.parse_args() |
| 62 | 62 |
|
| DTI_PID/WebServer/symbol_training/train.py | ||
|---|---|---|
| 69 | 69 |
if not os.path.isdir(opt.saved_path): |
| 70 | 70 |
os.mkdir(save_dir) |
| 71 | 71 |
|
| 72 |
# 학습할 클래스들을 저장하고 인식 시 불러와 사용합니다. |
|
| 72 | 73 |
with open(os.path.join(opt.saved_path, name + "_info.info"), 'w') as stream: |
| 73 | 74 |
con = str(len(DOFTECH_CLASSES)) |
| 74 | 75 |
names = '\n'.join(DOFTECH_CLASSES) |
내보내기 Unified diff
