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

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
Next Next commit
update datasets/registry
  • Loading branch information
@Turoad
Turoad committed Jul 8, 2021
commit 99f09eb2db9f87fd57cc57817451cd4136ddcb2b
5 changes: 2 additions & 3 deletions lanedet/datasets/registry.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -32,16 +32,15 @@ def build_dataloader(split_cfg, cfg, is_train=True):
shuffle = True
else:
shuffle = False
batch_size=cfg.batch_size

dataset = build_dataset(split_cfg, cfg)

init_fn = partial(
worker_init_fn, seed=cfg.seed)

samples_per_gpu = batch_size#cfg.batch_size // cfg.gpus
samples_per_gpu = cfg.batch_size // cfg.gpus
data_loader = torch.utils.data.DataLoader(
dataset, batch_size = batch_size, shuffle = shuffle,
dataset, batch_size = cfg.batch_size, shuffle = shuffle,
num_workers = cfg.workers, pin_memory = False, drop_last = False,
collate_fn=partial(collate, samples_per_gpu=samples_per_gpu),
worker_init_fn=init_fn)
Expand Down