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

[datasets] Add IIIT HWS dataset #1199

Merged
merged 2 commits into from
May 17, 2023
Merged

Conversation

felixT2K
Copy link
Contributor

@felixT2K felixT2K commented May 16, 2023

This PR:

  • adds logic to load iiit hws dataset
  • add corresponding tests
  • add to documentation

requested by: @tobiascornille in #1049

dataset: Link

from doctr.datasets import IIITHWS
train_set = IIITHWS(img_folder="/path/to/iiit-hws/Images_90K_Normalized",
                   label_path="/path/to/IIIT-HWS-90K.txt",
                   train=True)
img, target = train_set[0]
test_set = IIITHWS(img_folder="/path/to/iiit-hws/Images_90K_Normalized",
                   label_path="/path/to/IIIT-HWS-90K.txt")
                   train=False)
img, target = test_set[0]

@felixdittrich92 felixdittrich92 added this to the 0.6.1 milestone May 16, 2023
@felixdittrich92 felixdittrich92 added topic: documentation Improvements or additions to documentation type: enhancement Improvement ext: tests Related to tests folder module: datasets Related to doctr.datasets ext: docs Related to docs folder labels May 16, 2023
@codecov
Copy link

codecov bot commented May 16, 2023

Codecov Report

Merging #1199 (986a1c8) into main (cc0ceee) will increase coverage by 0.03%.
The diff coverage is 96.00%.

@@            Coverage Diff             @@
##             main    #1199      +/-   ##
==========================================
+ Coverage   94.98%   95.01%   +0.03%     
==========================================
  Files         149      150       +1     
  Lines        6417     6442      +25     
==========================================
+ Hits         6095     6121      +26     
+ Misses        322      321       -1     
Flag Coverage Δ
unittests 95.01% <96.00%> (+0.03%) ⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files Coverage Δ
doctr/datasets/iiithws.py 95.83% <95.83%> (ø)
doctr/datasets/__init__.py 100.00% <100.00%> (ø)

... and 1 file with indirect coverage changes

Copy link
Collaborator

@odulcy-mindee odulcy-mindee left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @felixdittrich92 ! 👏

@felixdittrich92 felixdittrich92 merged commit be4e8a2 into mindee:main May 17, 2023
@felixT2K felixT2K deleted the IIIT-HWS-dataset branch July 3, 2023 20:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ext: docs Related to docs folder ext: tests Related to tests folder module: datasets Related to doctr.datasets topic: documentation Improvements or additions to documentation type: enhancement Improvement
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants