diff --git a/cfg/transport.data b/cfg/transport.data new file mode 100644 index 0000000000000000000000000000000000000000..8482ece550f80df18b069b1bae6b000f6202cd5c --- /dev/null +++ b/cfg/transport.data @@ -0,0 +1,15 @@ +cfg_model=yolo +classes=8 +gpus = 0 +num_workers = 8 +batch_size = 8 +img_size = 416 +multi_scale = True +epochs = 100 +train=./yolo_transport_train/anno/train.txt +valid=./yolo_transport_train/anno/train.txt +names=./cfg/transport.names +#finetune_model=./coco_model/yolov3_coco.pt +finetune_model = ./weights-yolov3-transport/latest_416.pt +lr_step = 10,20,30 +lr0 = 0.001 diff --git a/cfg/transport.names b/cfg/transport.names new file mode 100644 index 0000000000000000000000000000000000000000..a1f5baf5ad0936802728b4d1b06c344cd2b10d5a --- /dev/null +++ b/cfg/transport.names @@ -0,0 +1,8 @@ +bicycle +car +motorcycle +airplane +bus +train +truck +boat