Open aaditnaren opened 5 years ago
after changing the yolo.cfg file save it as something else like yolo_1c.cfg in the same directory, u will have to keep the original yolo.cfg unchanged
Check out this answer in stackoverflow- https://stackoverflow.com/a/58812151/6517340
Did it work out for you ?
After placing the .weights and .cfg (after editing it in accordance to my classes) files properly and also the images and their annotations, I get an error while running the code.
import cv2 from darkflow.net.build import TFNet import matplotlib.pyplot as plt
%config InlineBackend.figure_format = 'svg'
options = { 'model' : 'cfg/yolo.cfg', 'load' : 'bin/yolo.weights', "batch": 8, "epoch": 25, 'threshold' : 0.3, "annotation": "./annotations/", "dataset": "./images/" }
tfnet = TFNet (options) tfnet.train()
OUTPUT: Parsing ./cfg/yolo.cfg Parsing cfg/yolo.cfg Loading bin/yolo.weights ...
AssertionError Traceback (most recent call last)