개정판 b64cc3b5
issue #1366: fix label training
| DTI_PID/WebServer/symbol_training/train.py | ||
|---|---|---|
| 19 | 19 |
from src.vis_utils.vis_tool import visdom_bbox |
| 20 | 20 |
|
| 21 | 21 |
loss_data = {'X': [], 'Y': [], 'legend_U':['total', 'coord', 'conf', 'cls']}
|
| 22 |
#visdom = visdom.Visdom(port='8080')
|
|
| 22 |
#visdom = visdom.Visdom(port='8088')
|
|
| 23 | 23 |
|
| 24 | 24 |
# 형상 CLASS |
| 25 | 25 |
DOFTECH_CLASSES= ['gate', 'globe', 'butterfly', 'check', 'ball', 'relief', |
| ... | ... | |
| 44 | 44 |
parser.add_argument("--decay", type=float, default=0.0005)
|
| 45 | 45 |
parser.add_argument("--dropout", type=float, default=0.5)
|
| 46 | 46 |
parser.add_argument("--num_epoches", type=int, default=205)
|
| 47 |
parser.add_argument("--test_interval", type=int, default=10, help="Number of epoches between testing phases")
|
|
| 47 |
parser.add_argument("--test_interval", type=int, default=20, help="Number of epoches between testing phases")
|
|
| 48 | 48 |
parser.add_argument("--object_scale", type=float, default=1.0)
|
| 49 | 49 |
parser.add_argument("--noobject_scale", type=float, default=0.5)
|
| 50 | 50 |
parser.add_argument("--class_scale", type=float, default=1.0)
|
내보내기 Unified diff
