Skip to content

Commit d4b209b

Browse files
xinhuaguXinhua Gu
and
Xinhua Gu
authored
fix img target size mismatch issue during inference (#542)
* force 100x100 crop for inference * correct the order of the parameters of Resize and CenterCrop functions * format java code with gradlew fJ for passing build --------- Co-authored-by: Xinhua Gu <[email protected]>
1 parent bf9b407 commit d4b209b

File tree

3 files changed

+5
-3
lines changed

3 files changed

+5
-3
lines changed

footwear_classification/src/main/java/com/examples/Inference.java

+2
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
import ai.djl.modality.Classifications;
1919
import ai.djl.modality.cv.Image;
2020
import ai.djl.modality.cv.ImageFactory;
21+
import ai.djl.modality.cv.transform.CenterCrop;
2122
import ai.djl.modality.cv.transform.Resize;
2223
import ai.djl.modality.cv.transform.ToTensor;
2324
import ai.djl.modality.cv.translator.ImageClassificationTranslator;
@@ -65,6 +66,7 @@ public static void main(String[] args) throws ModelException, TranslateException
6566
Translator<Image, Classifications> translator =
6667
ImageClassificationTranslator.builder()
6768
.addTransform(new Resize(Models.IMAGE_WIDTH, Models.IMAGE_HEIGHT))
69+
.addTransform(new CenterCrop(Models.IMAGE_WIDTH, Models.IMAGE_HEIGHT))
6870
.addTransform(new ToTensor())
6971
.optApplySoftmax(true)
7072
.build();

footwear_classification/src/main/java/com/examples/Training.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ public static void main(String[] args) throws IOException, TranslateException {
7070
// metrics collect and report key performance indicators, like accuracy
7171
trainer.setMetrics(new Metrics());
7272

73-
Shape inputShape = new Shape(1, 3, Models.IMAGE_HEIGHT, Models.IMAGE_HEIGHT);
73+
Shape inputShape = new Shape(1, 3, Models.IMAGE_HEIGHT, Models.IMAGE_WIDTH);
7474

7575
// initialize trainer with proper input shape
7676
trainer.initialize(inputShape);

malicious-url-detector/src/main/java/com/example/FilterProxy.java

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@
2323
import java.util.ArrayList;
2424

2525
/**
26-
* FilterProxy class receives requests on a server socket, and handles requests on individual threads
27-
* looking for malicious url patterns.
26+
* FilterProxy class receives requests on a server socket, and handles requests on individual
27+
* threads looking for malicious url patterns.
2828
*/
2929
public class FilterProxy implements Runnable {
3030

0 commit comments

Comments
 (0)