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.

Sign up for GitHub

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

Add condlane #20

Merged
merged 15 commits into from
Sep 11, 2021
Merged

Add condlane #20

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
f397ff0
add condlane
Turoad Jun 8, 2021
8d6e022
update condlane
Turoad Jun 19, 2021
11a62a9
fix interface
Jun 19, 2021
8a963a3
fix
Jun 20, 2021
d3e23e8
minor fix
Turoad Jun 28, 2021
2c80bfe
update config & rename models
Turoad Jun 28, 2021
bd6a70f
add README.md for condlane
Turoad Jul 2, 2021
99f09eb
update datasets/registry
Turoad Jul 8, 2021
9faf0dc
remove some magic number
Turoad Jul 8, 2021
f16d2d3
Merge branch 'condlane' of https://github.com/Turoad/lanedet into con…
Turoad Jul 8, 2021
315ce63
add condlane resnet50
Turoad Jul 8, 2021
5891920
minor fix
Turoad Jul 8, 2021
3fa3b9b
update readme
Turoad Sep 7, 2021
2fa9fca
minor fix
Turoad Sep 10, 2021
28ef010
Merge branch 'main' into condlane
Turoad Sep 10, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Merge branch 'main' into condlane
  • Loading branch information
@Turoad
Turoad authored Sep 10, 2021
commit 28ef0106f1c866ca647456e505247fe60e148406
1 change: 1 addition & 0 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ Supported detectors:
- [x] [LaneATT](configs/laneatt)
- [x] [CondLane](configs/condlane)


## Installation
<!--
Please refer to [INSTALL.md](INSTALL.md) for installation.
Expand Down
4 changes: 4 additions & 0 deletions lanedet/datasets/base_dataset.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,11 @@ def __len__(self):

def __getitem__(self, idx):
data_info = self.data_infos[idx]
if not osp.isfile(data_info['img_path']):
raise FileNotFoundError('cannot find file: {}'.format(data_info['img_path']))

img = cv2.imread(data_info['img_path'])

img = img[self.cfg.cut_height:, :, :]
sample = data_info.copy()
sample.update({'img': img})
Expand Down
You are viewing a condensed version of this merge commit. You can view the full changes here.