개정판 256188f0
issue #000: fix ai training
| DTI_PID/WebServer/app/recognition/index.py | ||
|---|---|---|
| 60 | 60 |
data_path = os.path.join(os.path.dirname(os.path.realpath(__file__)) + '\\..\\..\\symbol_training\\Data\\', data['name']) |
| 61 | 61 |
|
| 62 | 62 |
boxes = test_doftech_all_images.get_symbol(imgs, data['name'], root_path=data_path, trained_model1=os.path.join(data_path, 'checkpoint', data['name'] + "_only_params.pth"), \ |
| 63 |
trained_model2=None, save=False)
|
|
| 63 |
trained_model2=None, save=True)
|
|
| 64 | 64 |
|
| 65 | 65 |
return jsonify({'symbol_box': boxes[0]})
|
| 66 | 66 |
|
| DTI_PID/WebServer/symbol_recognition/test_doftech_all_images.py | ||
|---|---|---|
| 76 | 76 |
model2 = model1 |
| 77 | 77 |
|
| 78 | 78 |
colors = pickle.load(open(os.path.dirname(os.path.realpath(__file__)) + "/src/pallete", "rb")) |
| 79 |
colors = colors + colors + colors |
|
| 79 | 80 |
|
| 80 | 81 |
img_list = imgs |
| 81 | 82 |
|
내보내기 Unified diff
