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

Dataset Refactor #1

Closed
wants to merge 24 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
24 commits
Select commit Hold shift + click to select a range
14f3bf9
Initial commit. With args file
deepakgopinath Apr 15, 2021
7d1dd2f
Clean up
deepakgopinath Apr 16, 2021
018668b
Moved args file
deepakgopinath Apr 17, 2021
c1b6d28
Added init.py
deepakgopinath Apr 17, 2021
eaabd72
Added Base class for dataset. Added gaze dataset, awareness dataset. WIP
deepakgopinath Apr 17, 2021
bb910b0
Add embed to check road img
deepakgopinath Apr 17, 2021
478594c
Added seg mask extraction
deepakgopinath Apr 17, 2021
ab102f1
Added optic flow retrieval. Added optic flow scale factor and pad consts
deepakgopinath Apr 17, 2021
0e23578
Parse gaze points from pandas df
deepakgopinath Apr 18, 2021
5692b4e
Parse gaze points from pandas df
deepakgopinath Apr 18, 2021
40c9e71
Parse gaze points from pandas df
deepakgopinath Apr 18, 2021
9fa75de
Awareness dataset. WIP
deepakgopinath Apr 18, 2021
6e73a7d
Awareness data. Added annotation
deepakgopinath Apr 18, 2021
934e0fc
Pairwise Gaze Dataset for consistency term. WIP
deepakgopinath Apr 18, 2021
913e48a
pairwise metadata creation
deepakgopinath Apr 18, 2021
28d2d4f
Removed embed.
deepakgopinath Apr 18, 2021
2171ebd
Clean up import. Simplify Base class. Added args for gaze data pkl cache
deepakgopinath Apr 18, 2021
a7345f0
Improved comments and documentation
deepakgopinath Apr 18, 2021
1f4fdcd
Assert bug fix
deepakgopinath Apr 18, 2021
e861bb4
bug fix
deepakgopinath Apr 18, 2021
9f5eb9a
CognitiveHeatMap --> CHM. Comment before code. Removed pyc
deepakgopinath Apr 21, 2021
5c26326
Updated function docs
deepakgopinath Apr 21, 2021
3d6d754
Updated dataset classes. Deleted temp test
deepakgopinath May 7, 2021
3090d78
updated args file
deepakgopinath May 7, 2021
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
16 changes: 16 additions & 0 deletions src/chm/__init__.py
@@ -0,0 +1,16 @@
"""Root package info"""

__version__ = "0.1.0"
__author__ = "TRI RAD team"
__license__ = "MIT"
__copyright__ = "Copyright (c) 2021-2021, TRI"
__homepage__ = "https://github.com/ToyotaResearchInstitute/att-aware"
__docs__ = "att-aware is a library for attended awareness estimation"
__long_docs__ = """
Official [PyTorch](https://pytorch.org/) implementation of _Cognitive HeatMap_
invented by the RAD Team at [Toyota Research Institute (TRI)](https://www.tri.global/),
in particular for _Cognitive Heatmap_:
[*Cognitive Heatmap: A Model for Driver Situational Awareness*],
*Deepak Gopinath, Guy Rosman, Simon Stent, Katsuya Terahata, Luke Fletcher, John Leonard*.

"""
Empty file added src/chm/configs/__init__.py
Empty file.