개정판 db5c22f1
issue #1366: fix name
Change-Id: I14d1655af121323b88ea95f87986aad978e6ecb1
| DTI_PID/WebServer/app/recognition/index.py | ||
|---|---|---|
| 59 | 59 | |
| 60 | 60 |
data_path = os.path.join(os.path.dirname(os.path.realpath(__file__)) + '\\..\\..\\symbol_training\\Data\\', data['name']) |
| 61 | 61 | |
| 62 |
boxes = test_doftech_all_images.get_symbol(imgs, root_path=data_path, trained_model1=os.path.join(data_path, 'checkpoint', data['name'] + "_only_params.pth"), \ |
|
| 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 | 63 |
trained_model2=os.path.dirname(os.path.realpath( |
| 64 | 64 |
__file__)) + '\\..\\..\\symbol_recognition\\MODEL\\doftech_all_class_only_params_opc.pth') |
| 65 | 65 | |
| DTI_PID/WebServer/symbol_recognition/test_doftech_all_images.py | ||
|---|---|---|
| 47 | 47 |
# x, y, width, height |
| 48 | 48 |
return [self.class_info, self.start_h, self.start_w, self.end_h - self.start_h, self.end_w - self.start_w, self.iou] |
| 49 | 49 | |
| 50 |
def get_symbol(imgs, root_path=None, trained_model1=None, trained_model2=None): |
|
| 50 |
def get_symbol(imgs, name=None, root_path=None, trained_model1=None, trained_model2=None):
|
|
| 51 | 51 |
global colors |
| 52 | 52 | |
| 53 | 53 |
parser = argparse.ArgumentParser("You Only Look Once: Unified, Real-Time Object Detection")
|
내보내기 Unified diff