Skip to content

Latest commit

 

History

History
9 lines (6 loc) · 291 Bytes

File metadata and controls

9 lines (6 loc) · 291 Bytes

wall classifiater model Train

Prepare.

  • A wall image
  • An image that is not a wall

Overview.

The training code in this repository shuffles the two datasets of wall images and non-wall images, enters the data loader, performs a convolutional operation, and creates a .pth file.