Home Explore Help
Sign In
RONCC
/
NVIDIA-GPU-Bootcamp-Training
Watch 1
Star 0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Tree: f7a9033ad1
Branches Tags
NVIDIA-GPU-B...
/
hpc_ai
/
ai_science_cfd
/
English
/
python
/
source_code
/
dataset.py

dataset.py 2.7 KB
History Raw

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354 
  1. # Copyright (c) 2012, NVIDIA CORPORATION. All rights reserved.
    2. 

  2.  #
    3. 

  3.  # Redistribution and use in source and binary forms, with or without
    4. 

  4.  # modification, are permitted provided that the following conditions
    5. 

  5.  # are met:
    6. 

  6.  #  * Redistributions of source code must retain the above copyright
    7. 

  7.  #    notice, this list of conditions and the following disclaimer.
    8. 

  8.  #  * Redistributions in binary form must reproduce the above copyright
    9. 

  9.  #    notice, this list of conditions and the following disclaimer in the
    10. 

  10.  #    documentation and/or other materials provided with the distribution.
    11. 

  11.  #  * Neither the name of NVIDIA CORPORATION nor the names of its
    12. 

  12.  #    contributors may be used to endorse or promote products derived
    13. 

  13.  #    from this software without specific prior written permission.
    14. 

  14.  #
    15. 

  15.  # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AS IS'' AND ANY
    16. 

  16.  # EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
    17. 

  17.  # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
    18. 

  18.  # PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR
    19. 

  19.  # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
    20. 

  20.  # EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
    21. 

  21.  # PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
    22. 

  22.  # PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
    23. 

  23.  # OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
    24. 

  24.  # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
    25. 

  25.  # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
    26. 

  26. 

  27. import gdown
    28. 

  28. import os
    29. 

  29. ## CFD TRAIN DATASET
    30. 

  30. url = 'https://drive.google.com/uc?id=0BzsbU65NgrSuZDBMOW93OWpsMHM&export=download'
    31. 

  31. output = '/workspace/python/jupyter_notebook/CFD/data/train.tfrecords'
    32. 

  32. gdown.download(url, output, quiet=False,proxy=None)
    33. 

  33. 

  34. ## CFD TEST DATASET
    35. 

  35. url = 'https://drive.google.com/uc?id=1WSJLK0cOQehixJ6Tf5k0eYDcb4RJ5mXv&export=download'
    36. 

  36. output = '/workspace/python/jupyter_notebook/CFD/data/test.tfrecords'
    37. 

  37. gdown.download(url, output, quiet=False,proxy=None)
    38. 

  38. 

  39. ## CFD CONV_SDF MODEL
    40. 

  40. url = 'https://drive.google.com/uc?id=1pfR0io1CZKvXArGk-nt2wciUoAN_6Z08&export=download'
    41. 

  41. output = '/workspace/python/jupyter_notebook/CFD/conv_sdf_model.h5'
    42. 

  42. gdown.download(url, output, quiet=False,proxy=None)
    43. 

  43. 

  44. ## CFD CONV MODEL
    45. 

  45. url = 'https://drive.google.com/uc?id=1rFhqlQnTkzIyZocjAxMffucmS3FDI0_j&export=download'
    46. 

  46. output = '/workspace/python/jupyter_notebook/CFD/conv_model.h5'
    47. 

  47. gdown.download(url, output, quiet=False,proxy=None)
    48. 

  48. 

  49. 

  50. ## CFD TEST Dataset
    51. 

  51. url = 'https://drive.google.com/uc?id=0BzsbU65NgrSuR2NRRjBRMDVHaDQ&export=download'
    52. 

  52. output = '/workspace/python/jupyter_notebook/CFD/data/computed_car_flow.zip'
    53. 

  53. gdown.cached_download(url, output, quiet=False,proxy=None,postprocess=gdown.extractall)
    54. 

  54.