Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

how to generate YOLO_small.ckpt file by runing train.py? #74

Open
zimurui opened this issue Jul 22, 2018 · 6 comments
Open

how to generate YOLO_small.ckpt file by runing train.py? #74

zimurui opened this issue Jul 22, 2018 · 6 comments

Comments

@zimurui
Copy link

zimurui commented Jul 22, 2018

No description provided.

@cenwfohdkue
Copy link

it won't generator .ckpt file in new tensorflow version. just load the .meta and .date.

@Youpretty
Copy link

it won't generator .ckpt file in new tensorflow version. just load the .meta and .date.
@hainingbaby hi, hainingbaby. I trained the model with my own dataset, and generated 3 files named "yolo-10.data-00000-of-00001", "yolo-10.index" and 'yolo-10.meta'. How can i load the .meta and .data file? The code below can only open one weight file. Please help, tank you!
parser.add_argument('--weights', default="yolo-10.data-00000-of-00001", type=str)

@supermanhuyu
Copy link

@Youpretty
you can test it by "python test.py --weight_dir pascal_voc/output/your_outputdir --weights yolo-10"

@ZhouQianang
Copy link

The new tensorflow generate three files(.meta; .index; .data-XXX-of-XXX) instead of .ckpt file. And you can use these files like before. Put them in weights:
yolo.meta
yolo.index
yolo.data-0000-of-0001
Now run: python test.py --weights yolo

@EmbugMaker
Copy link

I'm new. I had this problem while running "train.py."Has anyone ever encountered this BUG? How to solve this problem?
File "E:/CODEs/Nerual_Net_work_model/yolo_tensorflow-master/train.py", line 90, in train
train_timer.remain(step, self.max_iter))
ValueError: Unknown format code 'f' for object of type 'str'

@ZhouQianang
Copy link

I'm new. I had this problem while running "train.py."Has anyone ever encountered this BUG? How to solve this problem?
File "E:/CODEs/Nerual_Net_work_model/yolo_tensorflow-master/train.py", line 90, in train
train_timer.remain(step, self.max_iter))
ValueError: Unknown format code 'f' for object of type 'str'

I changed the code<<

log_str = '''{} Epoch: {}, Step: {}, Learning rate: {},'''
''' Loss: {:5.3f}\nSpeed: {:.3f}s/iter,'''
'''' Load: {:.3f}s/iter, Remain: {}'''.format(
datetime.datetime.now().strftime('%m-%d %H:%M:%S'),
self.data.epoch,
int(step),
round(self.learning_rate.eval(session=self.sess), 6),
loss,
train_timer.average_time,
load_timer.average_time,
train_timer.remain(step, self.max_iter))

to<<

log_str = '''{} Epoch: {}, Step: {}, Learning rate: {},Loss: {:5.3f}\n
Speed: {:.3f}s/iter,Load: {:.3f}s/iter, Remain: {}'''.format(
datetime.datetime.now().strftime('%m-%d %H:%M:%S'),
self.data.epoch,
int(step),
round(self.learning_rate.eval(session=self.sess), 6),
float(loss),
float(train_timer.average_time),
float(load_timer.average_time),
train_timer.remain(step, self.max_iter))

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

6 participants