def train_dir(nets, optim, optim2, dataloader, args):
global image_size, it, image_sizes
caffe.set_mode_gpu()
if args.debug:
image_sizes = [[416, 416]]
while True:
if it % 500 == 0:
image_size = image_sizes[random.randint(0, len(image_sizes) - 1)]
print(image_size)
#im = cv2.imread('/home/busta/data/90kDICT32px/background/n03085781_3427.jpg')
#try:
process_batch(nets, optim, optim2, image_size, args)
if it % valid_interval == 0:
validate(nets, dataloader, image_size = [416, 416], split_words=False)
#except:
# continue
评论列表
文章目录