Open
Description
Hey, I try to use theegacy-yolov2 branch
%load_ext tensorboard
#we need imgaug 0.4 for image augmentations to work properly, see https://stackoverflow.com/questions/62580797/in-colab-doing-image-data-augmentation-with-imgaug-is-not-working-as-intended
!pip uninstall -y imgaug && pip uninstall -y albumentations && pip install imgaug==0.4
!pip install --upgrade --no-cache-dir gdown
!git clone -b legacy-yolov2 https://github.com/AIWintermuteAI/aXeleRate.git
import sys
sys.path.append('/content/aXeleRate')
from axelerate import setup_training, setup_inference
and below is my config
config = {
"model":{
"type": "Detector",
"architecture": "MobileNet7_5",
"input_size": 224,
"anchors": [[[0.76120044, 0.57155991], [0.6923348, 0.88535553], [0.47163042, 0.34163313]]],
"labels": ["licence"],
"obj_thresh" : 0.7,
"iou_thresh" : 0.5,
"coord_scale" : 1.0,
"class_scale" : 1.0,
"object_scale" : 5.0,
"no_object_scale" : 1.0
},
"weights" : {
"full": "",
"backend": "imagenet"
},
"train" : {
"actual_epoch": 100,
"train_image_folder": "/content/MyDrive/MyDrive/ALPR/im",
"train_annot_folder": "/content/MyDrive/MyDrive/ALPR/images",
"train_times": 1,
"valid_image_folder": "/content/MyDrive/MyDrive/ALPR/validation",
"valid_annot_folder": "/content/MyDrive/MyDrive/ALPR/validation_annot",
"valid_times": 1,
"valid_metric": "recall",
"batch_size": 4,
"learning_rate": 1e-4,
"saved_folder": F"/content/MyDrive/MyDrive/ALPR/ww44p3_result",
"first_trainable_layer": "",
"augmentation": True,
"is_only_detect" : False
},
"converter" : {
"type": ["k210"]
}
}
However, I am seeing the following error.
https://colab.research.google.com/drive/1ZVO7KDgd_X1aEVd7Jh6gAFDNlQLxhCiJ#scrollTo=QTswjSxVRAaG
Metadata
Metadata
Assignees
Labels
No labels