Deep learning : Which model to use for custom and unusual object detection
I'm working on a Bacteria detection project.
My question is : is it releavant to use pretrained weights and only train a model on few last layers to detect unusual objects ? Or should i train the model from scratch ?
(i have arround 2000 images of bacterias and I did'nt use any data augmentation yet)
I tried to train from scratch yoloV3 on the tiny-yolo model but i have a min loss of 4...
Any suggestions ? Any prefered model ?
deep-learning object-detection yolo vgg-net
asked Nov 25 '18 at 12:55
antoine Mathuantoine Mathu
35
add a comment |
I'm working on a Bacteria detection project.
My question is : is it releavant to use pretrained weights and only train a model on few last layers to detect unusual objects ? Or should i train the model from scratch ?
(i have arround 2000 images of bacterias and I did'nt use any data augmentation yet)
I tried to train from scratch yoloV3 on the tiny-yolo model but i have a min loss of 4...
Any suggestions ? Any prefered model ?
deep-learning object-detection yolo vgg-net
asked Nov 25 '18 at 12:55
antoine Mathuantoine Mathu
35
add a comment |
I'm working on a Bacteria detection project.
My question is : is it releavant to use pretrained weights and only train a model on few last layers to detect unusual objects ? Or should i train the model from scratch ?
(i have arround 2000 images of bacterias and I did'nt use any data augmentation yet)
I tried to train from scratch yoloV3 on the tiny-yolo model but i have a min loss of 4...
Any suggestions ? Any prefered model ?
deep-learning object-detection yolo vgg-net
asked Nov 25 '18 at 12:55
antoine Mathuantoine Mathu
35
I'm working on a Bacteria detection project.
My question is : is it releavant to use pretrained weights and only train a model on few last layers to detect unusual objects ? Or should i train the model from scratch ?
(i have arround 2000 images of bacterias and I did'nt use any data augmentation yet)
I tried to train from scratch yoloV3 on the tiny-yolo model but i have a min loss of 4...
Any suggestions ? Any prefered model ?
deep-learning object-detection yolo vgg-net
deep-learning object-detection yolo vgg-net
asked Nov 25 '18 at 12:55
antoine Mathuantoine Mathu
35
asked Nov 25 '18 at 12:55
antoine Mathuantoine Mathu
35
asked Nov 25 '18 at 12:55
antoine Mathuantoine Mathu
35
asked Nov 25 '18 at 12:55
antoine Mathuantoine Mathu
35
asked Nov 25 '18 at 12:55
antoine Mathuantoine Mathu
35
35
add a comment |
add a comment |
1 Answer
1
active
oldest
votes
I advice not to train the model from scratch as it will take a lot of your time and compute power.
The darknet feature extractor of YOLO (darknet-19 in YOLOv2 and darknet-53 in the case of YOLOv3) are first trained on Imagenet dataset for getting a feature extractor which is a lengthy process.
You should use the feature extractor as it is and train the layers after it.
YOLOv2 should work fine at some reasonable speed if you don't want very precise bounding boxes, but if you cannot work with some localization errors and want tight bounding boxes, go with YOLOv3.
For more tranining guidelines follow this link. AlexeyAB has curated all the steps to be followed to train YOLO on custom dataset.
answered Nov 26 '18 at 4:10
Vipin SharmaVipin Sharma
166
add a comment |
Your Answer
StackExchange.ifUsing("editor", function () {
StackExchange.using("externalEditor", function () {
StackExchange.using("snippets", function () {
StackExchange.snippets.init();
});
});
}, "code-snippets");
StackExchange.ready(function() {
var channelOptions = {
tags: "".split(" "),
id: "1"
};
initTagRenderer("".split(" "), "".split(" "), channelOptions);
StackExchange.using("externalEditor", function() {
// Have to fire editor after snippets, if snippets enabled
if (StackExchange.settings.snippets.snippetsEnabled) {
StackExchange.using("snippets", function() {
createEditor();
});
}
else {
createEditor();
}
});
function createEditor() {
StackExchange.prepareEditor({
heartbeatType: 'answer',
autoActivateHeartbeat: false,
convertImagesToLinks: true,
noModals: true,
showLowRepImageUploadWarning: true,
reputationToPostImages: 10,
bindNavPrevention: true,
postfix: "",
imageUploader: {
brandingHtml: "Powered by u003ca class="icon-imgur-white" href="https://imgur.com/"u003eu003c/au003e",
contentPolicyHtml: "User contributions licensed under u003ca href="https://creativecommons.org/licenses/by-sa/3.0/"u003ecc by-sa 3.0 with attribution requiredu003c/au003e u003ca href="https://stackoverflow.com/legal/content-policy"u003e(content policy)u003c/au003e",
allowUrls: true
},
onDemand: true,
discardSelector: ".discard-answer"
,immediatelyShowMarkdownHelp:true
});
}
});
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
StackExchange.ready(
function () {
StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fstackoverflow.com%2fquestions%2f53467645%2fdeep-learning-which-model-to-use-for-custom-and-unusual-object-detection%23new-answer', 'question_page');
}
);
Post as a guest
Required, but never shown
1 Answer
1
active
oldest
votes
1 Answer
1
active
oldest
votes
active
oldest
votes
active
oldest
votes
I advice not to train the model from scratch as it will take a lot of your time and compute power.
The darknet feature extractor of YOLO (darknet-19 in YOLOv2 and darknet-53 in the case of YOLOv3) are first trained on Imagenet dataset for getting a feature extractor which is a lengthy process.
You should use the feature extractor as it is and train the layers after it.
YOLOv2 should work fine at some reasonable speed if you don't want very precise bounding boxes, but if you cannot work with some localization errors and want tight bounding boxes, go with YOLOv3.
For more tranining guidelines follow this link. AlexeyAB has curated all the steps to be followed to train YOLO on custom dataset.
answered Nov 26 '18 at 4:10
Vipin SharmaVipin Sharma
166
add a comment |
I advice not to train the model from scratch as it will take a lot of your time and compute power.
The darknet feature extractor of YOLO (darknet-19 in YOLOv2 and darknet-53 in the case of YOLOv3) are first trained on Imagenet dataset for getting a feature extractor which is a lengthy process.
You should use the feature extractor as it is and train the layers after it.
YOLOv2 should work fine at some reasonable speed if you don't want very precise bounding boxes, but if you cannot work with some localization errors and want tight bounding boxes, go with YOLOv3.
For more tranining guidelines follow this link. AlexeyAB has curated all the steps to be followed to train YOLO on custom dataset.
answered Nov 26 '18 at 4:10
Vipin SharmaVipin Sharma
166
add a comment |
I advice not to train the model from scratch as it will take a lot of your time and compute power.
The darknet feature extractor of YOLO (darknet-19 in YOLOv2 and darknet-53 in the case of YOLOv3) are first trained on Imagenet dataset for getting a feature extractor which is a lengthy process.
You should use the feature extractor as it is and train the layers after it.
YOLOv2 should work fine at some reasonable speed if you don't want very precise bounding boxes, but if you cannot work with some localization errors and want tight bounding boxes, go with YOLOv3.
For more tranining guidelines follow this link. AlexeyAB has curated all the steps to be followed to train YOLO on custom dataset.
answered Nov 26 '18 at 4:10
Vipin SharmaVipin Sharma
166
I advice not to train the model from scratch as it will take a lot of your time and compute power.
The darknet feature extractor of YOLO (darknet-19 in YOLOv2 and darknet-53 in the case of YOLOv3) are first trained on Imagenet dataset for getting a feature extractor which is a lengthy process.
You should use the feature extractor as it is and train the layers after it.
YOLOv2 should work fine at some reasonable speed if you don't want very precise bounding boxes, but if you cannot work with some localization errors and want tight bounding boxes, go with YOLOv3.
For more tranining guidelines follow this link. AlexeyAB has curated all the steps to be followed to train YOLO on custom dataset.
answered Nov 26 '18 at 4:10
Vipin SharmaVipin Sharma
166
edited Nov 26 '18 at 8:51
answered Nov 26 '18 at 4:10
Vipin SharmaVipin Sharma
166
answered Nov 26 '18 at 4:10
Vipin SharmaVipin Sharma
166
answered Nov 26 '18 at 4:10
Vipin SharmaVipin Sharma
166
166
add a comment |
add a comment |
Thanks for contributing an answer to Stack Overflow!
- Please be sure to answer the question. Provide details and share your research!
But avoid …
- Asking for help, clarification, or responding to other answers.
- Making statements based on opinion; back them up with references or personal experience.
To learn more, see our tips on writing great answers.
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
StackExchange.ready(
function () {
StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fstackoverflow.com%2fquestions%2f53467645%2fdeep-learning-which-model-to-use-for-custom-and-unusual-object-detection%23new-answer', 'question_page');
}
);
Post as a guest
Required, but never shown
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
Sign up or log in
StackExchange.ready(function () {
StackExchange.helpers.onClickDraftSave('#login-link');
});
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Sign up using Google
Sign up using Facebook
Sign up using Email and Password
Post as a guest
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
Required, but never shown
