Skip to content

Commit 36e9b10

Browse files
smurchingsueann
authored andcommitted
Switch to getting resources as streams instead of files (databricks#101)
Replace calls to getResource (which look for resource files only on the local disk) with calls to getResourceAsStream, which uses the classloader to search for resource files anywhere in the classpath.
1 parent e47b398 commit 36e9b10

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/test/scala/com/databricks/sparkdl/ImageUtilsSuite.scala

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@ object ImageUtilsSuite {
3232

3333
/** Read image data into a BufferedImage, then use our utility method to convert to a row image */
3434
def getImageRow(resourcePath: String): Row = {
35-
val resourceFilename = getClass.getResource(resourcePath).getFile
36-
val imageBuffer = ImageIO.read(new File(resourceFilename))
35+
val resourceStream = getClass.getResourceAsStream(resourcePath)
36+
val imageBuffer = ImageIO.read(resourceStream)
3737
ImageUtils.spImageFromBufferedImage(imageBuffer)
3838
}
3939

@@ -50,8 +50,8 @@ class ImageUtilsSuite extends FunSuite {
5050
test("Test spImage resize.") {
5151
def javaResize(imagePath: String, tgtWidth: Int, tgtHeight: Int): Row = {
5252
// Read BufferedImage directly from file
53-
val resourceFilename = getClass.getResource(imagePath).getFile
54-
val srcImg = ImageIO.read(new File(resourceFilename))
53+
val resourceStream = getClass.getResourceAsStream(imagePath)
54+
val srcImg = ImageIO.read(resourceStream)
5555
val tgtImg = new BufferedImage(tgtWidth, tgtHeight, srcImg.getType)
5656
// scaledImg is a java.awt.Image which supports drawing but not pixel lookup by index.
5757
val scaledImg = srcImg.getScaledInstance(tgtWidth, tgtHeight, Image.SCALE_AREA_AVERAGING)

0 commit comments

Comments
 (0)