from network.YoloV1 import YoloV1
from model.dection import dectionYolov1

voc_labels = ["aeroplane", "bicycle", "bird", "boat", "bottle",
    "bus", "car", "cat", "chair", "cow", "diningtable", "dog",
    "horse", "motorbike", "person", "pottedplant", "sheep", "sofa",
    "train", "tvmonitor"]

net = YoloV1((448, 448, 3))
file_list = ['dataset/yolo_test_data/image/bicycle.jpg','dataset/yolo_test_data/image/car.jpg']
dectionYolov1(net,file_list,voc_labels,r'model_saved/yolov1/model.ckpt')

from network.YoloV1 import YoloV1
from model.dection import dectionYolov1

voc_labels = ["aeroplane", "bicycle", "bird", "boat", "bottle",
    "bus", "car", "cat", "chair", "cow", "diningtable", "dog",
    "horse", "motorbike", "person", "pottedplant", "sheep", "sofa",
    "train", "tvmonitor"]

net = YoloV1((448, 448, 3))
file_list = ['dataset/yolo_test_data/image/bicycle.jpg','dataset/yolo_test_data/image/car.jpg']
dectionYolov1(net,file_list,voc_labels,r'model_saved/yolov1/model.ckpt')
