Commit 3665f649 authored by josh's avatar josh

fixed little bug

parent 9c1f1f1b
......@@ -15,9 +15,7 @@ python Scripts/initDB.py ../ ImageDatabase
## Directory Structure Result:
```
├── ImageDatabase
│   ├── MeanShift
│   ├── Original
│   └── Sliced
│   └── Images
```
## "importAndProcessImage.py"
......@@ -35,81 +33,17 @@ python Scripts/importAndProcessImage.py TestResults/MS/GeoEye_MS_Original.jpg ..
## Database Directory Result:
```
├── MeanShift
│   └── GeoEye-0
Images
├── 0-jpg-GeoEye
│   ├── 0-jpg-GeoEye-Original.jpg
│   └── Sliced
│   ├── 128
│   │   ├── GeoEye-0-0-color-seg.jpg
│   │   ├── GeoEye-0-0-labels-image.jpg
│   │   ├── GeoEye-0-128-color-seg.jpg
│   │   ├── GeoEye-0-128-labels-image.jpg
│   │   ├── GeoEye-0-256-color-seg.jpg
│   │   ├── GeoEye-0-256-labels-image.jpg
│   │   ├── GeoEye-0-384-color-seg.jpg
│   │   ├── GeoEye-0-384-labels-image.jpg
│   │   ├── GeoEye-128-0-color-seg.jpg
│   │   ├── GeoEye-128-0-labels-image.jpg
│   │   ├── GeoEye-128-128-color-seg.jpg
│   │   ├── GeoEye-128-128-labels-image.jpg
│   │   ├── GeoEye-128-256-color-seg.jpg
│   │   ├── GeoEye-128-256-labels-image.jpg
│   │   ├── GeoEye-128-384-color-seg.jpg
│   │   ├── GeoEye-128-384-labels-image.jpg
│   │   ├── GeoEye-256-0-color-seg.jpg
│   │   ├── GeoEye-256-0-labels-image.jpg
│   │   ├── GeoEye-256-128-color-seg.jpg
│   │   ├── GeoEye-256-128-labels-image.jpg
│   │   ├── GeoEye-256-256-color-seg.jpg
│   │   ├── GeoEye-256-256-labels-image.jpg
│   │   ├── GeoEye-256-384-color-seg.jpg
│   │   ├── GeoEye-256-384-labels-image.jpg
│   │   ├── GeoEye-384-0-color-seg.jpg
│   │   ├── GeoEye-384-0-labels-image.jpg
│   │   ├── GeoEye-384-128-color-seg.jpg
│   │   ├── GeoEye-384-128-labels-image.jpg
│   │   ├── GeoEye-384-256-color-seg.jpg
│   │   ├── GeoEye-384-256-labels-image.jpg
│   │   ├── GeoEye-384-384-color-seg.jpg
│   │   └── GeoEye-384-384-labels-image.jpg
│   ├── 256
│   │   ├── GeoEye-0-0-color-seg.jpg
│   │   ├── GeoEye-0-0-labels-image.jpg
│   │   ├── GeoEye-0-256-color-seg.jpg
│   │   ├── GeoEye-0-256-labels-image.jpg
│   │   ├── GeoEye-256-0-color-seg.jpg
│   │   ├── GeoEye-256-0-labels-image.jpg
│   │   ├── GeoEye-256-256-color-seg.jpg
│   │   └── GeoEye-256-256-labels-image.jpg
│   └── 512
│   ├── GeoEye-0-0-color-seg.jpg
│   └── GeoEye-0-0-labels-image.jpg
├── Original
│   └── GeoEye-0.jpg
└── Sliced
└── GeoEye-0
├── 128
│   ├── GeoEye-0-0.jpg
│   ├── GeoEye-0-128.jpg
│   ├── GeoEye-0-256.jpg
│   ├── GeoEye-0-384.jpg
│   ├── GeoEye-128-0.jpg
│   ├── GeoEye-128-128.jpg
│   ├── GeoEye-128-256.jpg
│   ├── GeoEye-128-384.jpg
│   ├── GeoEye-256-0.jpg
│   ├── GeoEye-256-128.jpg
│   ├── GeoEye-256-256.jpg
│   ├── GeoEye-256-384.jpg
│   ├── GeoEye-384-0.jpg
│   ├── GeoEye-384-128.jpg
│   ├── GeoEye-384-256.jpg
│   └── GeoEye-384-384.jpg
├── 256
│   ├── GeoEye-0-0.jpg
│   ├── GeoEye-0-256.jpg
│   ├── GeoEye-256-0.jpg
│   └── GeoEye-256-256.jpg
└── 512
└── GeoEye-0-0.jpg
│   │   ├── 0-jpg-GeoEye-128-0-0
│   │   │   ├── 0-jpg-GeoEye-128-0-0-Sliced.jpg
│   │   │   └── Processed
│   │   │   ├── 0-jpg-GeoEye-128-0-0-MeanShiftLabels-Processed.npy
│   │   │   └── 0-jpg-GeoEye-128-0-0-MeanShiftSegmentation-Processed.jpg
```
## "sliceImage.py" program:
......
......@@ -24,7 +24,6 @@ def initDB( dbLoc: str, name: str ) -> bool:
File Struture:
|-- ImageDatabase
│   |-- Data
'''
dbLoc = pathlib.Path( dbLoc )
dbDir = dbLoc.joinpath( name )
......@@ -32,16 +31,11 @@ def initDB( dbLoc: str, name: str ) -> bool:
return False
else:
imagesDir = dbDir.joinpath( "Images" )
#slicedImagesDir = dbDir.joinpath( "Sliced" )
#meanShiftSegDir = dbDir.joinpath( "MeanShift" )
os.mkdir( dbDir )
os.mkdir( imagesDir )
#os.mkdir( slicedImagesDir )
#os.mkdir( meanShiftSegDir )
return True
def importAndProcess( imgPath: str, dbPath: str, newImgName: str ):
'''
Slices and segments an image and imports it into the database
......@@ -70,7 +64,10 @@ def importAndProcess( imgPath: str, dbPath: str, newImgName: str ):
# Auto increment id
dbIDs = [int( f.name.split( "-" )[0] ) for f in imagesPath.iterdir() ]
dbID = max( dbIDs ) + 1
dbID = 0
if len( dbIDs ) != 0:
dbID = max( dbIDs ) + 1
# Create dir for image
baseImagePath = imagesPath.joinpath( "{}-{}-{}".format( dbID, fileExt, newImgName ) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment