Skip to content

Commit

Permalink
Don't load weights.
Browse files Browse the repository at this point in the history
We are training from scratch. See
Viredery#27 (comment).
  • Loading branch information
khatchad committed Aug 1, 2024
1 parent dba9b11 commit af4c8f4
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions train_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,9 +69,6 @@
# %%
_ = model((batch_imgs, batch_metas), training=False)

# %%
model.load_weights('weights/faster_rcnn.h5', by_name=True)

# %%
proposals = model.simple_test_rpn(img, img_meta)
res = model.simple_test_bboxes(img, img_meta, proposals)
Expand Down

0 comments on commit af4c8f4

Please sign in to comment.